diff --git a/scripts/extra/convert_gt_yolo.py b/scripts/extra/convert_gt_yolo.py index c08c0b056..4affc5d29 100644 --- a/scripts/extra/convert_gt_yolo.py +++ b/scripts/extra/convert_gt_yolo.py @@ -55,11 +55,11 @@ def convert_yolo_coordinates_to_voc(x_c_n, y_c_n, width_n, height_n, img_width, image_name = tmp_file.split(".txt",1)[0] #print(image_name) ## check if image exists - for fname in os.listdir('../images'): + for fname in os.listdir('../images-optional'): if fname.startswith(image_name): ## image found #print(fname) - img = cv2.imread('../images/' + fname) + img = cv2.imread('../images-optional/' + fname) ## get image width and height img_height, img_width = img.shape[:2] break