赞
踩
错误原因:
解决办法:
数据集中所有的照片虽然都是.jpg格式的,但是有的照片没有进行转码,只是更改了后缀,因此在模型加载的时候才会报错。
错误的原因正是因为后缀名和图片的实际格式不符,才会导致imread读入为空
可以使用PIL工具将所有的jpg图片转换成为jpg图片,从而避免这样的麻烦。
转换代码-注意会删除源PNG文件,谨慎操作!
from PIL import Image import os def convert_png_to_jpg(folder_path): # 遍历文件夹中的所有文件 for filename in os.listdir(folder_path): if filename.endswith('.png'): # 构造输入和输出文件的完整路径 input_path = os.path.join(folder_path, filename) output_path = os.path.splitext(input_path)[0] + '.jpg' # 打开PNG图像并保存为JPEG格式 image = Image.open(input_path) image.convert('RGB').save(output_path, 'JPEG') print(f'Converted {input_path} to {output_path}') # 删除源文件 os.remove(input_path) print(f'Deleted {input_path}') # 指定包含PNG图像的文件夹路径 folder_path = '/content/demo2/images/' # 调用函数进行转换 convert_png_to_jpg(folder_path)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。