alfred8995 commited on
Commit
d888ad7
·
verified ·
1 Parent(s): 1d0e27a

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. miner.py +2 -2
miner.py CHANGED
@@ -74,8 +74,8 @@ class Miner:
74
  self.input_shape = self.session.get_inputs()[0].shape
75
 
76
  # Your export is fixed-size 1280, but we still read actual ONNX input shape first.
77
- self.input_height = self._safe_dim(self.input_shape[2], default=960)
78
- self.input_width = self._safe_dim(self.input_shape[3], default=960)
79
 
80
  # Tuned for validator scoring: reduce FP (FALSE_POSITIVE pillar),
81
  # preserve recall (MAP50, RECALL), improve precision.
 
74
  self.input_shape = self.session.get_inputs()[0].shape
75
 
76
  # Your export is fixed-size 1280, but we still read actual ONNX input shape first.
77
+ self.input_height = self._safe_dim(self.input_shape[2], default=1280)
78
+ self.input_width = self._safe_dim(self.input_shape[3], default=1280)
79
 
80
  # Tuned for validator scoring: reduce FP (FALSE_POSITIVE pillar),
81
  # preserve recall (MAP50, RECALL), improve precision.