|
@@ -24,7 +24,7 @@ def build_rtdetr(args, cfg, num_classes=80, trainable=False, deploy=False):
|
|
|
topk = 300,
|
|
topk = 300,
|
|
|
deploy = deploy,
|
|
deploy = deploy,
|
|
|
no_multi_labels = args.no_multi_labels,
|
|
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
|
|
nms_class_agnostic = args.nms_class_agnostic
|
|
|
)
|
|
)
|
|
|
|
|
|