Python：检验PS标注是否准确

主音贝斯手

于 2020-06-29 10:34:09 发布

阅读量193

点赞数

分类专栏：代码文章标签：神经网络

本文链接：https://blog.csdn.net/m0_37908025/article/details/107015238

版权

代码专栏收录该内容

11 篇文章 0 订阅

订阅专栏

import numpy as np
import os
import cv2 as cv

label_dir = './png/'
label_process_dir = './label_process/'
for file in os.listdir(label_dir):
    label = label_dir + file
    if os.path.isfile(label):
        print(file)
        # 获得一张图片的所有种类的灰度值
        label_array = cv.imread(label, 0)
        class_image = np.unique(np.array(label_array))
        print('class_image:', class_image)

        for i in class_image:
            if i in [255, 106, 0]:  # 84, 111, 208,
                pass
            else:
                print('error---------------------------------------------------------------')