批量删除纯黑图片及对应标签

import os
import shutil

import PIL.Image as Image
import cv2
import numpy as np
import pandas as pd

def selet_pic(imgpath,labpath):
    pic_path = os.listdir(imgpath)
    list = []
    for alldir in pic_path:
        child = os.path.join(imgpath,alldir)
        img = Image.open(child)
        # img.show()
        pic_arrays = np.array(img)   #将图片转化成数组
        if np.mean(pic_arrays) == 0:
            # print("图片为黑色",alldir)
            list.append(alldir)
        # print(list)
    return list


def del_pic(list,img_del_path,lab_del_path):
    for i in list:
        img_path = os.path.join(imgpath,i)
        lab_path = os.path.join(labpath,i)
        print(lab_path)
        os.remove(img_path)        #直接删除
        os.remove(lab_path)
        # shutil.move(img_path, img_del_path)
        # shutil.move(lab_path,lab_del_path)         #移到指定位置



if __name__== "__main__":
    imgpath = r'E:\1\img'
    labpath = r'E:\1\lab'
    list = selet_pic(imgpath,labpath)
    # del_pic(list,img_del_path,lab_del_path)
    del_pic(list)