Spaces:
Sleeping
Sleeping
yusef commited on
Commit ·
5d0fd2e
1
Parent(s): 415860e
Increase DETECTIONS_PER_IMAGE to 500 per tile
Browse files- model_manager.py +2 -2
model_manager.py
CHANGED
|
@@ -60,8 +60,8 @@ def load_model():
|
|
| 60 |
# NMS: Aggressively remove overlapping detections (lower = stricter)
|
| 61 |
cfg.MODEL.ROI_HEADS.NMS_THRESH_TEST = 0.3
|
| 62 |
|
| 63 |
-
# Max detections per image (
|
| 64 |
-
cfg.TEST.DETECTIONS_PER_IMAGE =
|
| 65 |
|
| 66 |
_predictor = DefaultPredictor(cfg)
|
| 67 |
_model_info = {
|
|
|
|
| 60 |
# NMS: Aggressively remove overlapping detections (lower = stricter)
|
| 61 |
cfg.MODEL.ROI_HEADS.NMS_THRESH_TEST = 0.3
|
| 62 |
|
| 63 |
+
# Max detections per image (رفعنا الحد عشان نكشف كل المباني في المناطق الكثيفة)
|
| 64 |
+
cfg.TEST.DETECTIONS_PER_IMAGE = 500
|
| 65 |
|
| 66 |
_predictor = DefaultPredictor(cfg)
|
| 67 |
_model_info = {
|