SuperBitDev commited on
Commit
c0bd005
·
verified ·
1 Parent(s): a4ea9c5

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. miner.py +3 -3
  2. 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=1280)
72
- self.input_width = self._safe_dim(self.input_shape[3], default=1280)
73
 
74
- self.conf_thres = 0.07
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:31ee30943eda5c04c67b76a612dff96dd6f6fdb44ea7ea28dd425493ce63a5ab
3
- size 19153171
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e2308ea9e8784fa26f57b93fe4d66837b07f031f9371e7f988aec0e8d7c9da9
3
+ size 19404973