python检查data图片集和label标签集是否相同

#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,yolo,图片,path,python,utf-8)