update
Browse files
miner.py
CHANGED
|
@@ -250,7 +250,7 @@ class Miner:
|
|
| 250 |
soft_sigma: float = 0.5,
|
| 251 |
) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
|
| 252 |
"""
|
| 253 |
-
Per-class Weighted Box Fusion with soft-NMS scoring.
|
| 254 |
|
| 255 |
For each cluster of overlapping boxes (IoU >= iou_thresh):
|
| 256 |
- Coords: confidence-weighted mean (more robust than picking top)
|
|
|
|
| 250 |
soft_sigma: float = 0.5,
|
| 251 |
) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
|
| 252 |
"""
|
| 253 |
+
Per-class WBF (Weighted Box Fusion) with soft-NMS scoring.
|
| 254 |
|
| 255 |
For each cluster of overlapping boxes (IoU >= iou_thresh):
|
| 256 |
- Coords: confidence-weighted mean (more robust than picking top)
|