#!/usr/bin/env python
# coding=utf-8
import os
import shutil
def getimagepath(path):
file=open(path)
data=file.read()
lines=data.replace(',',' ').replace('\t',' ').split('\n')
lines=[[ v.strip() for v in line.split(' ') if v.strip()!=''] for line in lines if len(line)>0 and line[0]!='#']
res=[line[0] for line in lines if line[1]=='1']
return res
cwd=os.getcwd()
print cwd
txtpath=cwd+'/ImageSets/Main/'
print txtpath
txtlist=os.listdir(txtpath)
namelist=[]
#print txtlist
for name in txtlist:
index=name.find('_')
if index!=-1:
namelist.append(name[:index])
namelist=list(set(namelist))
print len(namelist),'\n',namelist
newfolds=cwd+'/VOC'
oldfolds=cwd+'/JPEGImages'
if not os.path.exists(newfolds):
os.mkdir(newfolds)
datasets=['train','trainval','val']
for ds in datasets:
if not os.path.exists(newfolds+'/'+ds):
os.mkdir(newfolds+'/'+ds)
for name in namelist:
if not os.path.exists(newfolds+'/'+ds+'/'+name):
os.mkdir(newfolds+'/'+ds+'/'+name)
train_imagepath=getimagepath(txtpath+'/'+name+'_'+ds+'.txt')
if len(train_imagepath)<=0:
print "error in getimagepath ",txtpath+'/'+name+'_'+ds+'.txt'
exit(0)
for image in train_imagepath:
shutil.copyfile(oldfolds+'/'+image+'.jpg',newfolds+'/'+ds+'/'+name+'/'+image+'.jpg')
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
