fix error

extract_01.py

@@ -19,8 +19,8 @@ svo_list = sorted(svo_list)
 
 
 for svo in svo_list:
-  if svo[-5] == '1':
-    os.system('python3 /sources/depthsensing.py -c /sources/cfg.cfg -w /sources/mailsys.weights -m /sources/data.data -s ' + target_dir + svo + ' -o ' + target_dir + svo[:4] + '.csv')
+  if svo[-5] == '1':   
+    os.system('python3 /sources/depthsensing.py -c /sources/cfg.cfg -w /sources/mailsys.weights -m /sources/data.data -s ' + target_dir + svo + ' -o ' + target_dir + svo[:5] + '.csv')
   else:
     continue
 

extract_02.py

@@ -19,9 +19,9 @@ svo_list = sorted(svo_list)
 
 for svo in svo_list:
   if svo[-5] == '2':
-    os.system('python3 /sources/depthsensing.py -c /sources/cfg.cfg -w /sources/mailsys.weights -m /sources/data.data -s ' + target_dir + svo + ' -o ' + target_dir + svo[:4] + '.csv')
+    os.system('python3 /sources/depthsensing.py -c /sources/cfg.cfg -w /sources/mailsys.weights -m /sources/data.data -s ' + target_dir + svo + ' -o ' + target_dir + svo[:5] + '.csv')
   else:
-    pass
+    continue
 
 full_list = os.listdir(target_dir)
 csv_2_list = [file for file in full_list if file.endswith("2.csv")]