yusef commited on
Commit
5d0fd2e
·
1 Parent(s): 415860e

Increase DETECTIONS_PER_IMAGE to 500 per tile

Browse files
Files changed (1) hide show
  1. 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 (fewer = less overlap)
64
- cfg.TEST.DETECTIONS_PER_IMAGE = 200
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 = {