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

最新推荐文章于 2025-12-02 19:01:55 发布

原创最新推荐文章于 2025-12-02 19:01:55 发布 · 183 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #pillow #深度学习

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

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))