在用caffe进行训练时,有时候发现Label文件中所含有的图片名字中,在图片库中竟然找不到图片,当然会导致训练出错。
所以用Python写了个脚本来解决这样的问题。
import sys
import os
import os.path as osp
def filter(path):
file=open(osp.join(path,'label.txt'))
while 1:
line=file.readline()
print(line)
if not line:
break
for root, dirs, list in os.walk(path):
pass
f1=open(osp.join(path,'label.txt'),'r')
words=dict(line.strip('\n').split(' ',1) for line in f1)
word=set(line.strip('\n') for line in list )
same={k for k in words.keys() & word}#if python2,use'words,viewkeys()'
f3=open(osp.join(path,'label1.txt'),'w')
for e in same:
output=str("".join(str(e) )+' '+"".join(str(words[e]) )+'\n')
f3.write(output)
filter(r'C:\Users\Administrator\Desktop\file')