Update test_image.py

no bgr to rgb
sunhawk2100 · May 23, 2018 · cd900fc · cd900fc
1 parent 5422e8f
commit cd900fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_image.py b/test_image.py
@@ -11,11 +11,11 @@
     mtcnn_detector = MtcnnDetector(pnet=pnet, rnet=rnet, onet=onet, min_face_size=24)
 
     img = cv2.imread("./test.jpg")
-    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    img_bg = 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
 
-    vision.vis_face(img2,bboxs,landmarks)
+    vision.vis_face(img_bg,bboxs,landmarks)