Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,8 @@ def greet(name):
|
|
29 |
pnet, rnet, onet = create_mtcnn_net(p_model_path=args.pnet_path, r_model_path=args.rnet_path,o_model_path=args.onet_path, use_cuda=args.use_cuda)
|
30 |
mtcnn_detector = MtcnnDetector(pnet=pnet, rnet=rnet, onet=onet, min_face_size=args.min_face_size,threshold=thresh)
|
31 |
img = cv2.imread(name)
|
32 |
-
|
|
|
33 |
p_bboxs, r_bboxs, bboxs, landmarks = mtcnn_detector.detect_face(img)
|
34 |
save_name = args.save_name
|
35 |
return vis_face(img_bg, bboxs, landmarks, MIN_FACE_SIZE, save_name)
|
|
|
29 |
pnet, rnet, onet = create_mtcnn_net(p_model_path=args.pnet_path, r_model_path=args.rnet_path,o_model_path=args.onet_path, use_cuda=args.use_cuda)
|
30 |
mtcnn_detector = MtcnnDetector(pnet=pnet, rnet=rnet, onet=onet, min_face_size=args.min_face_size,threshold=thresh)
|
31 |
img = cv2.imread(name)
|
32 |
+
b,g,r = cv2.split(img)
|
33 |
+
img_bg = cv2.merge([r,g,b])
|
34 |
p_bboxs, r_bboxs, bboxs, landmarks = mtcnn_detector.detect_face(img)
|
35 |
save_name = args.save_name
|
36 |
return vis_face(img_bg, bboxs, landmarks, MIN_FACE_SIZE, save_name)
|