fix error

extract_01.py

@@ -4,24 +4,29 @@ import glob
 import pandas as pd
 
 #ssd_dir = /data/36536536520201231/
-ssd_dir = sys.argv[1]
+ssd_dir = str(sys.argv[1])
 try:
   os.system('mv '+ssd_dir+ ' /hdd')
 except:
   pass
+
 target_dir = '/hdd' + ssd_dir[5:]
 
 
 svo_list = os.listdir(target_dir)
+svo_list = sorted(svo_list)
+
+
 
 for svo in svo_list:
-  if svo[-5] == 1:
+  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')
   else:
-    pass
+    continue
 
 full_list = os.listdir(target_dir)
 csv_1_list = [file for file in full_list if file.endswith("1.csv")]
+csv_1_list = sorted(csv_1_list)
 
 conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
 FRAME_COUNTER = 0

extract_02.py

@@ -14,20 +14,23 @@ target_dir = '/hdd' + ssd_dir[5:]
 
 
 svo_list = os.listdir(target_dir)
+svo_list = sorted(svo_list)
+
 
 for svo in svo_list:
-  if svo[-5] == 2:
+  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')
   else:
     pass
 
 full_list = os.listdir(target_dir)
-csv_1_list = [file for file in full_list if file.endswith("1.csv")]
+csv_2_list = [file for file in full_list if file.endswith("2.csv")]
+csv_2_list = sorted(csv_2_list)
 
 conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
 FRAME_COUNTER = 0
 N_OF_FRAME = 1800
-for x in csv_1_list:
+for x in csv_2_list:
   df = pd.read_csv(os.path.join(target_dir,x), header=0)
   df['frame'] = df['frame'] + FRAME_COUNTER
   conc_df = pd.concat([conc_df,df])