export step updated

source/extract_01.py

@@ -0,0 +1,32 @@
+import sys
+import os
+import glob
+import pandas as pd
+
+#ssd_dir = /data/36536536520201231/
+ssd_dir = sys.argv[1]
+os.system('mv '+ssd_dir+ ' /hdd')
+
+target_dir = '/hdd' + ssd_dir[5:]
+
+
+svo_list = os.listdir(target_dir)
+
+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')
+  else:
+    pass
+
+full_list = os.listdir(target_dir)
+csv_1_list = [file for file in full_list if file.endswith("1.csv")]
+
+conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
+FRAME_COUNTER = 0
+N_OF_FRAME = 1800
+for x in csv_1_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]
+  FRAME_COUNTER += N_OF_FRAME
+conc_df.to_csv(os.path.join(target_dir,'cam1.csv'), index=False)

source/extract_02.py

@@ -0,0 +1,32 @@
+import sys
+import os
+import glob
+import pandas as pd
+
+#ssd_dir = /data/36536536520201231/
+ssd_dir = sys.argv[1]
+os.system('mv '+ssd_dir+ ' /hdd')
+
+target_dir = '/hdd' + ssd_dir[5:]
+
+
+svo_list = os.listdir(target_dir)
+
+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')
+  else:
+    pass
+
+full_list = os.listdir(target_dir)
+csv_1_list = [file for file in full_list if file.endswith("1.csv")]
+
+conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
+FRAME_COUNTER = 0
+N_OF_FRAME = 1800
+for x in csv_1_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]
+  FRAME_COUNTER += N_OF_FRAME
+conc_df.to_csv(os.path.join(target_dir,'cam2.csv'), index=False)