Upload 2 files

miner.py

@@ -30,7 +30,7 @@ class Miner:
     model_class_names = ["bottle", "can", "cup"]
     _model_to_competition_cls = np.array([1, 2, 0], dtype=np.int32)
     input_size = 1280
-    iou_thres = 0.4
+    iou_thres = 0.3
     cross_iou_thresh = 0.7
     min_side = 8.0
     min_box_area = 100.0
@@ -526,3 +526,19 @@ class Miner:
                 )
             )
         return results
+
+# if __name__ == "__main__":
+
+# # predict batch with local images for testing
+#     import json
+#     from time import time
+
+#     test_images = [
+#         cv2.imread(str(p)) for p in sorted(Path("test_images").glob("*.jpg"))
+#     ]
+#     miner = Miner(Path("hf_repo"))
+#     start_time = time()
+#     results = miner.predict_batch(test_images, offset=0, n_keypoints=0)
+#     end_time = time()
+#     print(f"Predicted {len(test_images)} images in {end_time - start_time:.2f} seconds")
+#     print(json.dumps([r.dict() for r in results], indent=2))

weights.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a43473993494103558080b2f69741a91068b91c807f57d32120dee5c1201bd7c
-size 19358729
+oid sha256:e34c250d8bd2c1a4f05bdb58d1153ff5a3713d30c094ff10be682740ec85c8b1
+size 19358782