【格式转换】将JPEG图片批量处理为jpg格式

【yolov5训练错误】WARNING: Ignoring corrupted image_vibration_xu的博客-CSDN博客https://blog.csdn.net/vibration_xu/article/details/122580372

数据图片中有JPEG和jpg两种文件格式，train时会出现报错而无法正常训练，故需要将文件格式统一成.jpg格式后便可成功训练将JPEG等格式批量转化为jpg格式。

代码如下，注意路径中不应存在中文。

import os
import cv2 as cv

image_path = 'J:/102HorizontalProject/zong/image'    #设置图片读取路径
save_path = 'J:/yolov5-master-cow/data/images'    #设置图片保存路径

if not os.path.exists(save_path):    #判断路径是否正确，并打开
    os.makedirs(save_path)

image_file = os.listdir(image_path)

for image in image_file:
    if image.split('.')[-1] in ['bmp', 'jpg', 'jpeg', 'png', 'JPG', 'PNG']:
        str = image.rsplit(".", 1)    #从右侧判断是否有符号“.”，并对image的名称做一次分割。如112345.jpeg分割后的str为["112345","jpeg"]
        output_img_name = str[0] + ".jpg"    #取列表中的第一个字符串与“.jpg”放在一起。
        src = cv.imread(os.path.join(image_path, image))
        newimg = cv.imwrite(save_path + '/' + output_img_name, src)
print('FINISHED')