语义分割数据集批量删除纯黑背景的图片

GuiBridge

于 2024-04-25 10:32:45 发布

阅读量165

点赞数 3

CC 4.0 BY-SA版权

文章标签： python numpy pillow 深度学习

本文链接：https://blog.youkuaiyun.com/GuiBridge/article/details/138181921

本文介绍了使用Python的os和PIL库遍历目录中的JPEG和PNG图片，如果图片所有像素平均值为0（即全黑），则删除该图片及其对应的原始JPEG文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os 
import PIL.Image as Image
import numpy as np



origin_jpg_path = './JPEGImages/'
label_png_path = './SegmentationClass/'

i = 0
label_pngs = os.listdir(label_png_path)
for label_png in label_pngs:
    (label_png_name, pic_PNGorJPG) = os.path.splitext(label_png)
    jpg_name = label_png_name + '.jpg'
    png_path = os.path.join(label_png_path,label_png)
    jpg_path = os.path.join(origin_jpg_path,jpg_name)
    img = Image.open(png_path)
    pic_arrays = np.array(img) 
    if np.mean(pic_arrays) == 0:
        i = i + 1
        os.remove(png_path)
        os.remove(jpg_path)
        print("已删除{}张图片".format(i))