Upload folder using huggingface_hub
Browse files- miner.py +3 -3
- weights.onnx +2 -2
miner.py
CHANGED
|
@@ -68,10 +68,10 @@ class Miner:
|
|
| 68 |
self.input_shape = self.session.get_inputs()[0].shape
|
| 69 |
|
| 70 |
# Your export is fixed-size 1280, but we still read actual ONNX input shape first.
|
| 71 |
-
self.input_height = self._safe_dim(self.input_shape[2], default=
|
| 72 |
-
self.input_width = self._safe_dim(self.input_shape[3], default=
|
| 73 |
|
| 74 |
-
self.conf_thres = 0.
|
| 75 |
self.iou_thres = 0.6
|
| 76 |
self.max_det = 300
|
| 77 |
self.use_tta = True
|
|
|
|
| 68 |
self.input_shape = self.session.get_inputs()[0].shape
|
| 69 |
|
| 70 |
# Your export is fixed-size 1280, but we still read actual ONNX input shape first.
|
| 71 |
+
self.input_height = self._safe_dim(self.input_shape[2], default=960)
|
| 72 |
+
self.input_width = self._safe_dim(self.input_shape[3], default=960)
|
| 73 |
|
| 74 |
+
self.conf_thres = 0.01
|
| 75 |
self.iou_thres = 0.6
|
| 76 |
self.max_det = 300
|
| 77 |
self.use_tta = True
|
weights.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0e2308ea9e8784fa26f57b93fe4d66837b07f031f9371e7f988aec0e8d7c9da9
|
| 3 |
+
size 19404973
|