scorevision: push artifact
Browse files
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.
|
| 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.
|
| 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.
|