删掉加载出错的文件:
import PIL.Image as Image
import os
import warnings
def check_img_valid(img_path):
result = False
try:
Image.open(img_path)
result = True
except UserWarning as msg:
print(msg)
print(img_path)
return result
warnings.filterwarnings("error", category=UserWarning)
path = r'C:\xxxxx\data'
files = os.listdir(path)
for file in files:
new_path = os.path.join(path, file)
if not check_img_valid(new_path):
os.remove(new_path)
批量检查并删除无效图片文件
该代码段用于遍历指定路径下的文件,通过PIL库尝试打开每个文件。如果文件是有效的图片,则忽略;若文件打开失败,引发UserWarning,则打印警告信息并删除该文件,确保目录中只保留有效图片。
2073

被折叠的 条评论
为什么被折叠?



