attempt screw recognition

anno.txt

@@ -0,0 +1,24 @@
+/home/cole/item-sort/positives/IMG_10208.jpg 1 4 3 80 110
+/home/cole/item-sort/positives/IMG_1155.jpg 1 1 1 30 32
+/home/cole/item-sort/positives/IMG_1184.jpg 1 1 1 34 30
+/home/cole/item-sort/positives/IMG_1287.jpg 1 2 2 35 29
+/home/cole/item-sort/positives/IMG_13351.jpg 1 2 2 165 75
+/home/cole/item-sort/positives/IMG_13857.jpg 1 2 5 90 142
+/home/cole/item-sort/positives/IMG_1472.jpg 1 1 1 43 29
+/home/cole/item-sort/positives/IMG_1584.jpg 1 1 1 45 31
+/home/cole/item-sort/positives/IMG_1683.jpg 1 1 0 48 32
+/home/cole/item-sort/positives/IMG_1764.jpg 1 2 1 38 39
+/home/cole/item-sort/positives/IMG_1776.jpg 1 1 0 45 36
+/home/cole/item-sort/positives/IMG_19530.jpg 1 4 4 120 149
+/home/cole/item-sort/positives/IMG_2200.jpg 1 1 0 46 42
+/home/cole/item-sort/positives/IMG_2772.jpg 1 0 0 83 32
+/home/cole/item-sort/positives/IMG_2784.jpg 1 1 1 46 55
+/home/cole/item-sort/positives/IMG_2800.jpg 1 1 1 53 48
+/home/cole/item-sort/positives/IMG_3285.jpg 1 1 1 71 43
+/home/cole/item-sort/positives/IMG_3604.jpg 1 1 0 104 33
+/home/cole/item-sort/positives/IMG_4550.jpg 1 1 1 127 32
+
+/home/cole/item-sort/positives/IMG_4879.jpg 1 2 2 37 114
+/home/cole/item-sort/positives/IMG_7412.jpg 1 2 2 105 65
+/home/cole/item-sort/positives/IMG_7654.jpg 1 4 2 80 84
+/home/cole/item-sort/positives/IMG_7740.jpg 1 1 2 126 57

bg.txt

@@ -0,0 +1,38 @@
+negatives/IMG_10285.jpg
+negatives/IMG_1190.jpg
+negatives/IMG_1225.jpg
+negatives/IMG_1443.jpg
+negatives/IMG_1482.jpg
+negatives/IMG_17202.jpg
+negatives/IMG_18939.jpg
+negatives/IMG_19860.jpg
+negatives/IMG_2116.jpg
+negatives/IMG_2408.jpg
+negatives/IMG_2475.jpg
+negatives/IMG_2548.jpg
+negatives/IMG_2597.jpg
+negatives/IMG_2700.jpg
+negatives/IMG_27048.jpg
+negatives/IMG_2805.jpg
+negatives/IMG_28583.jpg
+negatives/IMG_30940.jpg
+negatives/IMG_35235.jpg
+negatives/IMG_3599.jpg
+negatives/IMG_36040.jpg
+negatives/IMG_36400.jpg
+negatives/IMG_3666.jpg
+negatives/IMG_3840.jpg
+negatives/IMG_3844.jpg
+negatives/IMG_40176.jpg
+negatives/IMG_42840.jpg
+negatives/IMG_5041.jpg
+negatives/IMG_5184.jpg
+negatives/IMG_5402.jpg
+negatives/IMG_5700.jpg
+negatives/IMG_5820.jpg
+negatives/IMG_6014.jpg
+negatives/IMG_6016.jpg
+negatives/IMG_6240.jpg
+negatives/IMG_6435.jpg
+negatives/IMG_7482.jpg
+negatives/IMG_9800.jpg

main.py

@@ -66,6 +66,8 @@ def swap(a, b):
 ap = argparse.ArgumentParser()
 ap.add_argument("-i", "--image", required=True,
                 help="path to the input image")
+ap.add_argument("-c", "--cascade", required=True,
+                help="path to the cascade")
 ap.add_argument("-w", "--width", type=float, required=True,
                 help="width of the left-most object in the image (in inches)")
 ap.add_argument("-n", "--number", type=int, required=False,
@@ -112,7 +114,18 @@ cnts = imutils.grab_contours(cnts)
 pixelsPerMetric = None
 num = 0
 
-
+cascade = cv2.CascadeClassifier()
+if not cascade.load(cv2.samples.findFile(args2.cascade)):
+    print('--(!)Error loading face cascade')
+    exit(0)
+screws = cascade.detectMultiScale(image)
+frame = image.copy()
+for (x,y,w,h) in screws:
+    center = (x + w//2, y + h//2)
+    frame = cv2.ellipse(frame, center, (w//2, h//2), 0, 0, 360, (255, 0, 255), 4)
+if args2.show:
+    cv2.imshow("Item Sorter", frame)
+    cv2.waitKey(0)
 # Calibration loop
 for c in cnts:
     # if the contour is not sufficiently large, ignore it