zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix relative path for VOC download (#880) 

Cleaned up code
This commit is contained in:
NanoCode012 2020-09-01 06:06:10 +07:00 committed by GitHub
parent 3e04d20c7d
commit 901243c780
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
data/scripts/get_voc.sh
View File

@ -163,48 +163,29 @@ f = open('../tmp/train.txt', 'r')
lines = f.readlines() lines = f.readlines()
for line in lines: for line in lines:
#print(line.split('/')[-1][:-1]) line = "/".join(line.split('/')[-5:]).strip()
line = "/".join(line.split('/')[2:]) if (os.path.exists("../" + line)):
#print(line) os.system("cp ../"+ line + " ../VOC/images/train")
if (os.path.exists("../" + line[:-1])):
os.system("cp ../"+ line[:-1] + " ../VOC/images/train")
print(os.path.exists('../tmp/train.txt'))
f = open('../tmp/train.txt', 'r')
lines = f.readlines()
for line in lines:
#print(line.split('/')[-1][:-1])
line = "/".join(line.split('/')[2:])
line = line.replace('JPEGImages', 'labels') line = line.replace('JPEGImages', 'labels')
line = line.replace('jpg', 'txt') line = line.replace('jpg', 'txt')
#print(line) if (os.path.exists("../" + line)):
if (os.path.exists("../" + line[:-1])): os.system("cp ../"+ line + " ../VOC/labels/train")
os.system("cp ../"+ line[:-1] + " ../VOC/labels/train")
print(os.path.exists('../tmp/2007_test.txt')) print(os.path.exists('../tmp/2007_test.txt'))
f = open('../tmp/2007_test.txt', 'r') f = open('../tmp/2007_test.txt', 'r')
lines = f.readlines() lines = f.readlines()
for line in lines: for line in lines:
#print(line.split('/')[-1][:-1]) line = "/".join(line.split('/')[-5:]).strip()
line = "/".join(line.split('/')[2:]) if (os.path.exists("../" + line)):
os.system("cp ../"+ line + " ../VOC/images/val")
if (os.path.exists("../" + line[:-1])):
os.system("cp ../"+ line[:-1] + " ../VOC/images/val")
print(os.path.exists('../tmp/2007_test.txt'))
f = open('../tmp/2007_test.txt', 'r')
lines = f.readlines()
for line in lines:
#print(line.split('/')[-1][:-1])
line = "/".join(line.split('/')[2:])
line = line.replace('JPEGImages', 'labels') line = line.replace('JPEGImages', 'labels')
line = line.replace('jpg', 'txt') line = line.replace('jpg', 'txt')
#print(line) if (os.path.exists("../" + line)):
if (os.path.exists("../" + line[:-1])): os.system("cp ../"+ line + " ../VOC/labels/val")
os.system("cp ../"+ line[:-1] + " ../VOC/labels/val")
END END