这里删除的为宽高不为50的图片
#!/usr/bin/python
# -*- coding: utf-8 -*-
import os
from PIL import Image
import glob
path = "/home/li/data/data(RGB)/train/aug_gt/3/"
paths = glob.glob(os.path.join(path, '*.jpg'))
# 输出所有文件和文件夹
for file in paths:
fp = open(file, 'rb')
img = Image.open(fp)
fp.close()
width = img.size[0]
height = img.size[1]
if (width !=50) or (height !=50):
os.remove(file)