Changed bat_detector logic

detector.py

@@ -75,16 +75,29 @@ def bat_detector(img):
     # pathlib.PosixPath = pathlib.WindowsPath
     learn = load_learner("model.pkl")
 
-    for bat in cropped_bats:
+    # for bat in cropped_bats:
+    #     with learn.no_bar(), learn.no_logging():
+    #         label, _, probs = learn.predict(bat)
+    #     p1=f"{probs[0]:.4f}"
+    #     p2=f"{probs[1]:.4f}"
+    #     if label == '!bat' and p1 > '0.5':
+    #         pass
+    #     elif label == 'bat' and p2 > '0.5':
+    #         totalBats += 1
+    #     else:
+    #         pass
+    def predict(bat):
+        labels = ("!bat", "bat")
         with learn.no_bar(), learn.no_logging():
-            label, _, probs = learn.predict(bat)
-        p1=f"{probs[0]:.4f}"
-        p2=f"{probs[1]:.4f}"
-        if label == '!bat' and p1 > '0.5':
-            pass
-        elif label == 'bat' and p2 > '0.5':
+            _, _, probs = learn.predict(bat)
+        return (tuple(zip(labels, map(lambda x: f"{x:.4f}", probs))))
+
+    results = [predict(bat) for bat in cropped_bats]
+
+
+
+    for result in results:
+        if result[0][1] < '0.5' and result[1][1] > '0.75':
             totalBats += 1
-        else:
-            pass
-    print((time.time()-start))
-    return totalBats
+        print((time.time()-start))
+        return totalBats