Skip to content

Instantly share code, notes, and snippets.

@46bit

46bit/face_detect.py

Created January 31, 2015 02:33
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save 46bit/d49f6fd44b9e690a6ac5 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save 46bit/d49f6fd44b9e690a6ac5 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. 46bit created this gist Jan 31, 2015.
    45 changes: 45 additions & 0 deletions face_detect.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,45 @@
    import sys, cv2

    # Refactored https://realpython.com/blog/python/face-recognition-with-python/

    def cascade_detect(cascade, image):
    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    return cascade.detectMultiScale(
    gray_image,
    scaleFactor = 1.15,
    minNeighbors = 5,
    minSize = (30, 30),
    flags = cv2.cv.CV_HAAR_SCALE_IMAGE
    )

    def detections_draw(image, detections):
    for (x, y, w, h) in detections:
    print "({0}, {1}, {2}, {3})".format(x, y, w, h)
    cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)

    def main(argv = None):
    if argv is None:
    argv = sys.argv

    cascade_path = sys.argv[1]
    image_path = sys.argv[2]
    result_path = sys.argv[3] if len(sys.argv) > 3 else None

    cascade = cv2.CascadeClassifier(cascade_path)
    image = cv2.imread(image_path)
    if image is None:
    print "ERROR: Image did not load."
    return 2

    detections = cascade_detect(cascade, image)
    detections_draw(image, detections)

    print "Found {0} objects!".format(len(detections))
    if result_path is None:
    cv2.imshow("Objects found", image)
    cv2.waitKey(0)
    else:
    cv2.imwrite(result_path, image)

    if __name__ == "__main__":
    sys.exit(main())