import os
from PIL import Image
def check_and_delete_images(folder_path):
for subdir in os.listdir(folder_path):
subdir_path = os.path.join(folder_path, subdir)
if os.path.isdir(subdir_path):
for filename in os.listdir(subdir_path):
if filename.endswith(".jpg") or filename.endswith(".jpeg"):
file_path = os.path.join(subdir_path, filename)
try:
with Image.open(file_path) as img:
# 检查图像是否为空
if img.size[0] == 0 or img.size[1] == 0:
print(f"{filename}: 图像为空,将被删除")
# os.remove(file_path)
else:
# print(f"{filename}: 图像正常")
continue
except (IOError, SyntaxError) as e:
print(f"{filename}: 图像损坏,将被删除,错误信息:{str(e)}")
# os.remove(file_path)
# 替换为你的文件夹路径
folder_path = "/root/"+"autodl-tmp/yolov5/classify/caoyaodatas/train"
check_and_delete_images(folder_path)
05-08
314
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
06-06
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交