mailsys_server/extract_02.py

import sys
import os
import glob
import pandas as pd

#ssd_dir = /data/36536536520201231/
target_dir = sys.argv[1]


svo_list = os.listdir(target_dir)
svo_list = sorted(svo_list)
del svo_list[-1]

for svo in svo_list:
  if svo[-5:] == '2.svo':
    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

full_list = os.listdir(target_dir)
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_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])
  FRAME_COUNTER += N_OF_FRAME
conc_df.to_csv(os.path.join(target_dir,'cam2.csv'), index=False)