From 901243c7806be07b31073440cf721e73532a0734 Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue, 1 Sep 2020 06:06:10 +0700
Subject: [PATCH] Fix relative path for VOC download (#880)

Cleaned up code
---
 data/scripts/get_voc.sh | 43 ++++++++++++-----------------------------
 1 file changed, 12 insertions(+), 31 deletions(-)

diff --git a/data/scripts/get_voc.sh b/data/scripts/get_voc.sh
index 94843a6..f34467a 100644
--- a/data/scripts/get_voc.sh
+++ b/data/scripts/get_voc.sh
@@ -163,48 +163,29 @@ f = open('../tmp/train.txt', 'r')
 lines = f.readlines()
 
 for line in lines:
-    #print(line.split('/')[-1][:-1])
-    line = "/".join(line.split('/')[2:])
-    #print(line)
-    if (os.path.exists("../" + line[:-1])):
-        os.system("cp ../"+ line[:-1] + " ../VOC/images/train")
+    line = "/".join(line.split('/')[-5:]).strip()
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../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('jpg', 'txt')
-    #print(line)
-    if (os.path.exists("../" + line[:-1])):
-        os.system("cp ../"+ line[:-1] + " ../VOC/labels/train")
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/labels/train")
+
 
 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:])
-    
-    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 = "/".join(line.split('/')[-5:]).strip()
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/images/val")
+        
     line = line.replace('JPEGImages', 'labels')
     line = line.replace('jpg', 'txt')
-    #print(line)
-    if (os.path.exists("../" + line[:-1])):
-        os.system("cp ../"+ line[:-1] + " ../VOC/labels/val")
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/labels/val")
 
 END