Enderfga commited on
Commit
91c1fb4
·
1 Parent(s): 3c9f33b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- img_bg = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 
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)