From 01555ccf7a1048e8d304db98970e98de2077890f Mon Sep 17 00:00:00 2001
From: Sergey <smska87@bk.ru>
Date: Wed, 4 Jun 2025 12:10:48 +0300
Subject: [PATCH] fixed passing command line arguments to the model (conf, nms)

---
 tools/eval.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/eval.py b/tools/eval.py
index 2940ca1..5e7d2e6 100644
--- a/tools/eval.py
+++ b/tools/eval.py
@@ -94,7 +94,9 @@ def main():
 
     model = build_local_model(config, device)
     model.head.nms = True
-
+    model.head.nms_conf_thre = args.conf
+    model.head.nms_iou_thre = args.nms
+    
     model.cuda(args.local_rank)
     model.eval()