test: yolov11_box_inference 3type_box_detection #22

boostcampaitech7 · Nov 16, 2024 · b70229b · b70229b
1 parent 705899c
commit b70229b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/box_detect/yolov11_inference.py b/box_detect/yolov11_inference.py
@@ -17,4 +17,4 @@
                 image_path = os.path.join(patient_dir, image_file)
                 print(f"Running inference on: {image_path}")
                 # Run inference and save results
-                model.predict(image_path, save=True, imgsz=2048, conf=0.5,max_det=3,augment=True)
+                model.predict(image_path, save=True, imgsz=2048,iou=0.3, conf=0.3,max_det=3,augment=True)