#encoding:utf-8
#检查label和data数据集是否相同,若不相同输出不同的部分
def cheak_data_label:
import os
py_path=os.getcwd()
print "now , python file is located in " + py_path
label_path = py_path+"/label" #获取label文件夹位置
#print "label path:" + label_path
data_path = py_path+"/label" #获取data文件夹位置
label_files = os.listdir(label_path) #获取label文件夹下的文件名称
data_files = os.listdir(data_path) #获取data文件夹下的文件名称
L = []
for l in label_files:#检验label文件夹下的文件在data中是否存在
tmp = l.split(".")
L.append(tmp[0])#L是存储所有的label文件中的数字部分
#print tmp[0]
D=[]
for d in data_files:
tmp = d.split(".")
D.append(tmp[0])
print "exit in data but not in label:"
for d_n in D:
if d_n not in L:
print d_n
print "exit in label but not in data:"
for l_n in L:
if l_n not in D:
print l_n
print "complete!"
if __name__ == '__main__':
cheak_data_label()
python检查data图片集和label标签集是否相同
最新推荐文章于 2023-04-19 17:58:21 发布