--- /dev/null
+import numpy as np\r
+import cv2, cv\r
+import common\r
+\r
+def detect(img, cascade):\r
+ min_size = (20, 20)\r
+ haar_scale = 1.1\r
+ min_neighbors = 3\r
+ haar_flags = 0\r
+ rects = cascade.detectMultiScale(img, haar_scale, min_neighbors, haar_flags, min_size)\r
+ if len(rects) == 0:\r
+ return\r
+ rects[:,2:] += rects[:,:2]\r
+ return rects\r
+\r
+def detect_turned(img, cascade):\r
+ img_t = cv2.transpose(img)\r
+ img_cw = cv2.flip(img_t, 1)\r
+ img_ccw = cv2.flip(img_t, 0)\r
+ r = detect(img, cascade)\r
+ r_cw = detect(img_cw, cascade)\r
+ r_ccw = detect(img_ccw, cascade)\r
+\r
+ h, w = img.shape[:2]\r
+ if r_cw is not None:\r
+ r_cw[:,[0, 2]] = h - r_cw[:,[0, 2]] - 1\r
+ r_cw = r_cw[:,[1,0,3,2]]\r
+ if r_ccw is not None:\r
+ r_ccw[:,[1, 3]] = w - r_ccw[:,[1, 3]] - 1\r
+ r_ccw = r_ccw[:,[1,0,3,2]]\r
+ rects = np.vstack( [a for a in [r, r_cw, r_ccw] if a is not None] )\r
+ return rects\r
+\r
+def process_image(fn, cascade):\r
+ pass\r
+ \r
+ \r
+\r
+if __name__ == '__main__':\r
+ import sys\r
+ import getopt\r
+ args, img_mask = getopt.getopt(sys.argv[1:], '', ['cascade='])\r
+ args = dict(args)\r
+ cascade_fn = args.get('--cascade', "../../data/haarcascades/haarcascade_frontalface_alt.xml")\r
+\r
+ cascade = cv2.CascadeClassifier(cascade_fn)\r
+\r
+\r
+ img = cv2.imread('test.jpg')\r
+ h, w = img.shape[:2]\r
+ r = 512.0 / max(h, w)\r
+ small = cv2.resize(img, (int(w*r), int(h*r)), interpolation=cv2.INTER_AREA)\r
+ rects = detect_turned(small, cascade)\r
+ print rects\r
+ for x1, y1, x2, y2 in rects:\r
+ cv2.rectangle(small, (x1, y1), (x2, y2), (0, 255, 0))\r
+ cv2.circle(small, (x1, y1), 2, (0, 0, 255), -1)\r
+\r
+\r
+\r
+ cv2.imshow('img', small)\r
+ cv2.waitKey()\r
+\r
+ \r
+\r
+'''\r
+\r
+\r
+ img = cv2.imread('test.jpg')\r
+ h, w = img.shape[:2]\r
+\r
+\r
+ r = 512.0 / max(h, w)\r
+ small = cv2.resize(img, (w*r, h*r), interpolation=cv2.INTER_AREA)\r
+\r
+cv2.imshow('img', small)\r
+cv2.waitKey()\r
+\r
+'''
\ No newline at end of file