Add "magic sorting" feature

detect.py

@@ -319,6 +319,9 @@ def detect(calibration_width, img_file, show, quick):
         # cv2.putText(orig, "{:.5f}in".format(itemhr),
         #	(int(trbrX + 20), int(trbrY)), cv2.FONT_HERSHEY_SIMPLEX,
         #	0.65, (255, 255, 255), 2)
+        if objtype == "Unknown":
+            objtype = magicSort(c)
+            output = "{:.2f}in".format(itemw) + " x {:.2f}in".format(itemh)
         if circular:
             cv2.putText(orig, str(objtype),
                         (int(x - 25), int(y + radius + 20)
@@ -331,8 +334,7 @@ def detect(calibration_width, img_file, show, quick):
                         0.6, (50, 50, 220), 2)
         output = ""
         objname = objtype;
-        if objtype == "Unknown":
-            output = "{:.2f}in".format(itemw) + " x {:.2f}in".format(itemh)
+        
         if objtype == "Screw" or objtype == "Standoff":
             output = str(iteml) + "in"
             objname += str(iteml)
@@ -359,3 +361,19 @@ def detect(calibration_width, img_file, show, quick):
         cv2.waitKey(0)
     return (list, orig)
         
+def magicSort(contour):
+    moments = cv2.moments(contour)
+    humoments = cv2.HuMoments(moments)
+    humoments[6] = abs(humoments[6]) #it's possible for the last number to change sign if item is mirrored
+    magicNumber = 0
+    for i in range(0,7):
+	    humoments[i] = -1 * math.copysign(1.0, humoments[i]) * math.log10(abs(humoments[i]))
+        
+    for i in range(0,7):
+        magicNumber += humoments[i]
+        print(str(humoments[i]))
+    #print(magicNumber)
+    name = "Unknown: "
+    name += str(int(round(magicNumber[0] / 2) * 2))
+    print(name)
+    return name