今天一个目标检测的比赛中,需要读取的图片格式是,jpg,但数据集包含有部分png格式图片,在此我选用笨鸟方法,用os里面的工具实现批量的改写。
wd = getcwd() # D:\master\yolov4-main\yolov4-main
# **************************************************
# 先将文件夹里面的格式转为.jpg,后面只能读.jpg格式的图片
print(os.path.join(wd, 'model_data', 'helmet_image'))
fp1 = os.listdir(os.path.join(wd, 'model_data', 'helmet_image')) # 列表
num = len(fp1)
for i in range(num):
# print(fp1[i])
old_name = os.path.join(wd, 'model_data', 'helmet_image', fp1[i])
a, b = os.path.splitext(fp1[i])
# print(type(a))
if b != '.jpg':
new_name = os.path.join(wd, 'model_data', 'helmet_image', (a + '.jpg'))
if os.path.exists(new_name) is False:
os.renames(old_name, new_name)
os.path.join
那部分改成自己的文件的路径。当然也可使用opencv.write。