Merge pull request kuaikuaikim#13 from sakhalkark/patch-1

Use OpenCV color converter
zeroer71 · Mar 25, 2018 · 71bf183 · 71bf183
2 parents 4820428 + f1b82e0
commit 71bf183
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test_image.py b/test_image.py
@@ -11,8 +11,9 @@
     mtcnn_detector = MtcnnDetector(pnet=pnet, rnet=rnet, onet=onet, min_face_size=24)
 
     img = cv2.imread("./test.jpg")
-    b, g, r = cv2.split(img)
-    img2 = cv2.merge([r, g, b])
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    #b, g, r = cv2.split(img)
+    #img2 = cv2.merge([r, g, b])
 
     bboxs, landmarks = mtcnn_detector.detect_face(img)
     # print box_align