Procházet zdrojové kódy

we find NMS is beneficial

yjh0410 před 1 rokem
rodič
revize
c417365922
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      models/detectors/rtdetr/build.py

+ 1 - 1
models/detectors/rtdetr/build.py

@@ -24,7 +24,7 @@ def build_rtdetr(args, cfg, num_classes=80, trainable=False, deploy=False):
                     topk            = 300,
                     deploy          = deploy,
                     no_multi_labels = args.no_multi_labels,
-                    use_nms         = False,   # NMS is beneficial 
+                    use_nms         = True,   # NMS is beneficial 
                     nms_class_agnostic = args.nms_class_agnostic
                     )