计算图像分割mask的灰度级个数、以及删除空的分割数据（数据为自动驾驶车道线分割）

import os from tqdm import tqdm import cv2 import numpy as np def main(pth,im,ms): img_path = [os.path.join(pth,i) for i in os.listdir(pth)] for i in tqdm(img_path): j = i.replace('images','masks').replace(im,ms) tmp = cv2.imread(j) if len(np.unique(tmp)) ==1 : os.remove(i) os.remove(j) if __name__=='__main__': root = './data/images' # data 路径 img_fm = '.jpg' # data 的后缀偏移 mask_fm = '_mask.png' # mask 的后缀偏移 main(pth=root,im=img_fm,ms=mask_fm)