Update yolov8_det_trt.py

wang-xinyu · May 15, 2024 · b2c4306 · b2c4306
1 parent 32a44a1
commit b2c4306
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolov8/yolov8_det_trt.py b/yolov8/yolov8_det_trt.py
@@ -438,7 +438,7 @@ def run(self):
     try:
         print('batch size is', yolov8_wrapper.batch_size)
 
-        image_dir = "samples/"
+        image_dir = "images/"
         image_path_batches = get_img_path_batches(yolov8_wrapper.batch_size, image_dir)
 
         for i in range(10):