mailsys_server/extract_01.py

import sys
import os
import glob
import pandas as pd

#ssd_dir = /data/36536536520201231/
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':
    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:
    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
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)