coolroman commited on
Commit
c9f69ea
·
verified ·
1 Parent(s): a502e22

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. miner.py +2 -2
miner.py CHANGED
@@ -32,13 +32,13 @@ class Miner:
32
  # validator output order
33
  class_names = ["balaclava", "hoodie", "glove", "bat", "spray paint", "graffiti"]
34
  input_size = 1280
35
- iou_thres = 0.40
36
  cross_iou_thresh = 0.70
37
  max_aspect_ratio = 10.0
38
  max_det = 150
39
  # tuned on consensus-GT, healthy distribution: [balaclava, hoodie, glove, bat, spray, graffiti]
40
  _conf_thres_array = np.array(
41
- [0.60, 0.70, 0.70, 0.80, 0.50, 0.20], dtype=np.float32
42
  )
43
  # per-class rescue gap: if a class has no box above threshold, admit top-1
44
  # if its score >= (threshold - bonus). Inspired by new-vmodel0's _bonus_array.
 
32
  # validator output order
33
  class_names = ["balaclava", "hoodie", "glove", "bat", "spray paint", "graffiti"]
34
  input_size = 1280
35
+ iou_thres = 0.30
36
  cross_iou_thresh = 0.70
37
  max_aspect_ratio = 10.0
38
  max_det = 150
39
  # tuned on consensus-GT, healthy distribution: [balaclava, hoodie, glove, bat, spray, graffiti]
40
  _conf_thres_array = np.array(
41
+ [0.30, 0.70, 0.70, 0.50, 0.50, 0.40], dtype=np.float32
42
  )
43
  # per-class rescue gap: if a class has no box above threshold, admit top-1
44
  # if its score >= (threshold - bonus). Inspired by new-vmodel0's _bonus_array.