Selaa lähdekoodia

we find NMS is beneficial

yjh0410 1 vuosi sitten
vanhempi
sitoutus
c417365922
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  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
                     )