mailsys_server/source/extract_01.py

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)