Upload folder using huggingface_hub
Browse files
miner.py
CHANGED
|
@@ -68,8 +68,8 @@ 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.01
|
| 75 |
self.iou_thres = 0.6
|
|
|
|
| 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=1280)
|
| 72 |
+
self.input_width = self._safe_dim(self.input_shape[3], default=1280)
|
| 73 |
|
| 74 |
self.conf_thres = 0.01
|
| 75 |
self.iou_thres = 0.6
|