Explorar o código

we find NMS is beneficial

yjh0410 hai 1 ano
pai
achega
c417365922
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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
                     )