khaidir · September 7, 2015 06:45 · Sep 5, 2012 · Sep 5, 2012
diff --git a/cv2_detect_demo.py b/cv2_detect_demo.py
@@ -0,0 +1,34 @@
+import time
+from cv2_detect import detect
+import cv2
+import cv2.cv as cv
+
+
+def draw_rects(img, rects, color):
+    for x1, y1, x2, y2 in rects:
+        cv2.rectangle(img, (x1, y1), (x2, y2), color, 2)
+
+
+def demo(in_fn, out_fn):
+    print ">>> Loading image..."
+    img_color = cv2.imread(in_fn)
+    img_gray = cv2.cvtColor(img_color, cv.CV_RGB2GRAY)
+    img_gray = cv2.equalizeHist(img_gray)
+    print in_fn, img_gray.shape
+
+    print ">>> Detecting faces..."
+    start = time.time()
+    rects = detect(img_gray)
+    end = time.time()
+    print 'time:', end - start
+    img_out = img_color.copy()
+    draw_rects(img_out, rects, (0, 255, 0))
+    cv2.imwrite(out_fn, img_out)
+
+
+def main():
+    demo('pic.jpg', 'pic.detect.jpg')
+
+
+if __name__ == '__main__':
+    main()
diff --git a/cv2_detect_demo.py.stdout b/cv2_detect_demo.py.stdout
@@ -0,0 +1,4 @@
+>>> Loading image...
+pic.jpg (2400, 3600)
+>>> Detecting faces...
+time: 8.11453986168
diff --git a/haarcascades/haarcascade_eye.xml b/haarcascades/haarcascade_eye.xml
diff --git a/haarcascades/haarcascade_eye_tree_eyeglasses.xml b/haarcascades/haarcascade_eye_tree_eyeglasses.xml
diff --git a/haarcascades/haarcascade_frontalface_alt.xml b/haarcascades/haarcascade_frontalface_alt.xml
diff --git a/haarcascades/haarcascade_frontalface_alt2.xml b/haarcascades/haarcascade_frontalface_alt2.xml
diff --git a/haarcascades/haarcascade_frontalface_alt_tree.xml b/haarcascades/haarcascade_frontalface_alt_tree.xml
diff --git a/haarcascades/haarcascade_frontalface_default.xml b/haarcascades/haarcascade_frontalface_default.xml
diff --git a/haarcascades/haarcascade_fullbody.xml b/haarcascades/haarcascade_fullbody.xml
diff --git a/haarcascades/haarcascade_lefteye_2splits.xml b/haarcascades/haarcascade_lefteye_2splits.xml
diff --git a/haarcascades/haarcascade_lowerbody.xml b/haarcascades/haarcascade_lowerbody.xml
diff --git a/haarcascades/haarcascade_mcs_eyepair_big.xml b/haarcascades/haarcascade_mcs_eyepair_big.xml
diff --git a/haarcascades/haarcascade_mcs_eyepair_small.xml b/haarcascades/haarcascade_mcs_eyepair_small.xml
diff --git a/haarcascades/haarcascade_mcs_leftear.xml b/haarcascades/haarcascade_mcs_leftear.xml
diff --git a/haarcascades/haarcascade_mcs_lefteye.xml b/haarcascades/haarcascade_mcs_lefteye.xml
diff --git a/haarcascades/haarcascade_mcs_mouth.xml b/haarcascades/haarcascade_mcs_mouth.xml
diff --git a/haarcascades/haarcascade_mcs_nose.xml b/haarcascades/haarcascade_mcs_nose.xml
diff --git a/haarcascades/haarcascade_mcs_rightear.xml b/haarcascades/haarcascade_mcs_rightear.xml
diff --git a/haarcascades/haarcascade_mcs_righteye.xml b/haarcascades/haarcascade_mcs_righteye.xml
diff --git a/haarcascades/haarcascade_mcs_upperbody.xml b/haarcascades/haarcascade_mcs_upperbody.xml
diff --git a/haarcascades/haarcascade_profileface.xml b/haarcascades/haarcascade_profileface.xml
diff --git a/haarcascades/haarcascade_righteye_2splits.xml b/haarcascades/haarcascade_righteye_2splits.xml
diff --git a/haarcascades/haarcascade_upperbody.xml b/haarcascades/haarcascade_upperbody.xml
diff --git a/pic.detect.jpg b/pic.detect.jpg
diff --git a/pic.jpg b/pic.jpg
diff --git a/cv2_detect.py b/cv2_detect.py
@@ -0,0 +1,16 @@
+import cv2
+import cv2.cv as cv
+
+
+def detect(img, cascade_fn='haarcascades/haarcascade_frontalface_alt.xml',
+           scaleFactor=1.3, minNeighbors=4, minSize=(20, 20),
+           flags=cv.CV_HAAR_SCALE_IMAGE):
+
+    cascade = cv2.CascadeClassifier(cascade_fn)
+    rects = cascade.detectMultiScale(img, scaleFactor=scaleFactor,
+                                     minNeighbors=minNeighbors,
+                                     minSize=minSize, flags=flags)
+    if len(rects) == 0:
+        return []
+    rects[:, 2:] += rects[:, :2]
+    return rects
No results found