将label全黑的删除，合并将于label序号相同的data一起删除。

Fly*Boy

于 2020-07-22 16:01:26 发布

阅读量155

点赞数

本文链接：https://blog.csdn.net/mvpboy123/article/details/107515672

版权

import os.path
import shutil
import glob
import cv2
root = "/home/zhengsc/Desktop/qiepian/Task100_Colon/new_label/"
img_path = glob.glob(root  + "*.png")
num = len(img_path)
print(num)
n = 0
for i in range(num):
    img = cv2.imread(img_path[i])

    if 255 in img:  #黑白图像，白色255，如果全黑为0，如果不是则含有255
        # print(img_path[i])
        n = n + 1
    else:
        data = os.path.join(img_path[i][0:44] + "new_data/" + img_path[i][54:])
        # print(data)
        os.remove(data)
        os.remove(img_path[i])
        print("delete image:"+ img_path[i])
print(n)