Преглед на файлове

we find NMS is beneficial

yjh0410 преди 1 година
родител
ревизия
c417365922
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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
                     )