소스 검색

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
                     )