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

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. miner.py +2 -2
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=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
 
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