Tomislav115 commited on
Commit
03486f6
·
verified ·
1 Parent(s): a1cffa4

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. miner.py +2 -2
miner.py CHANGED
@@ -71,8 +71,8 @@ class Miner:
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
78
 
 
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.15
75
+ self.iou_thres = 0.5
76
  self.max_det = 300
77
  self.use_tta = True
78