check if agnostic_nms in settings

supervisely-ecosystem · Apr 26, 2023 · 9e56dfb · 9e56dfb
1 parent a76b283
commit 9e56dfb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/supervisely/serve/src/main.py b/supervisely/serve/src/main.py
@@ -144,8 +144,12 @@ def predict(self, image_path: str, settings: Dict[str, Any]) -> List[sly.nn.Pred
         inf_out = self.model(img, augment=augment)[0]
 
         # Apply NMS
+        if "agnostic_nms" in settings:
+            is_agnostic = settings["agnostic_nms"]
+        else:
+            is_agnostic = False
         output = non_max_suppression(
-            inf_out, conf_thres=conf_thres, iou_thres=iou_thres, agnostic=settings["agnostic_nms"]
+            inf_out, conf_thres=conf_thres, iou_thres=iou_thres, agnostic=is_agnostic
         )
 
         predictions = []