Update infer.py

manickavela29 · Mar 13, 2024 · 3516949 · 3516949
1 parent ae362e4
commit 3516949
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Detection/Gold-YOLO/tools/infer.py b/Detection/Gold-YOLO/tools/infer.py
@@ -100,7 +100,7 @@ def run(weights=osp.join(ROOT, 'yolov6s.pt'),
             os.makedirs(save_txt_path)
 
     # Inference
-    inferer = Inferer(source, weights, device, yaml, img_size, half)
+    inferer = Inferer(source, None, None, weights, device, yaml, img_size, half)
     inferer.infer(conf_thres, iou_thres, classes, agnostic_nms, max_det, save_dir, save_txt, not not_save_img, hide_labels, hide_conf, view_img)
 
     if save_txt or not not_save_img: