From f5c486cb7f10b70138d272d459b16b646df8ce05 Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Thu, 14 Nov 2024 12:00:24 -0300
Subject: [PATCH] Remove NMS of RT-DETR

---
 config_infer_primary_rtdetr.txt | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/config_infer_primary_rtdetr.txt b/config_infer_primary_rtdetr.txt
index 111eda3..80f86f1 100644
--- a/config_infer_primary_rtdetr.txt
+++ b/config_infer_primary_rtdetr.txt
@@ -13,7 +13,7 @@ interval=0
 gie-unique-id=1
 process-mode=1
 network-type=0
-cluster-mode=2
+cluster-mode=4
 maintain-aspect-ratio=0
 #workspace-size=2000
 parse-bbox-func-name=NvDsInferParseYolo
@@ -22,6 +22,5 @@ custom-lib-path=nvdsinfer_custom_impl_Yolo/libnvdsinfer_custom_impl_Yolo.so
 engine-create-func-name=NvDsInferYoloCudaEngineGet
 
 [class-attrs-all]
-nms-iou-threshold=0.45
 pre-cluster-threshold=0.25
 topk=300