import sys import os import glob import pandas as pd #ssd_dir = /data/36536536520201231/ ssd_dir = sys.argv[1] try: os.system('mv '+ssd_dir+ ' /hdd') except: pass 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)