Increased maximal number of detections during NMS, added logging for adaptive router scores

Author: MihinP

partinet/DynamicDet/models/yolo.py

@@ -240,7 +240,7 @@ def forward_once(self, x, profile=False):
                 self.get_score = False
             if self.get_score:
                 return score
-
+        print(f"This micrograph has difficulty score {score} for threshold {self.dy_thres}")
         need_second = self.training or (not self.dynamic) or score[:, 0] < self.dy_thres
         need_first_head = self.training or (self.dynamic and score[:, 0] >= self.dy_thres)
 

partinet/DynamicDet/utils/general.py

@@ -613,7 +613,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
 
     # Settings
     min_wh, max_wh = 2, 4096  # (pixels) minimum and maximum box width and height
-    max_det = 300  # maximum number of detections per image
+    max_det = 1000  # maximum number of detections per image
     max_nms = 30000  # maximum number of boxes into torchvision.ops.nms()
     time_limit = 10.0  # seconds to quit after
     redundant = True  # require redundant detections