--- a +++ b/src/subject_info.py @@ -0,0 +1,54 @@ +# create a folder for the registered subject + +import os +import shutil +import pickle + +SUBJECT_NAME = 'benny_cai' + +SCAN_POSE = 'side' +# SCAN_POSE = 'front' + +if __name__ == '__main__': + + folder_path = 'data/' + SUBJECT_NAME + if not os.path.isdir(folder_path): + os.mkdir(folder_path) + os.chdir(folder_path) + + if not os.path.isdir(SCAN_POSE): + os.mkdir(SCAN_POSE) + os.chdir(SCAN_POSE) + + for subfolder in ['color_images', + 'depth_images', + 'intrinsics', + 'OpenPose', + 'ViTPose_base', + 'ViTPose_large']: + if not os.path.isdir(subfolder): + os.mkdir(subfolder) + + # copy odometry.log from 'data' folder + src = '../../odometry.log' + dst = 'odometry.log' + shutil.copyfile(src, dst) + + file1 = open("../../subject_name.txt", "a") + file1.write(SUBJECT_NAME + '_' + SCAN_POSE + "\n") + file1.close() + + for subfolder in ['OpenPose', + 'ViTPose_base', + 'ViTPose_large']: + + for subsubfolder in ['keypoints', 'output_images']: + if not os.path.isdir(subfolder + '/' + subsubfolder): + os.mkdir(subfolder + '/' + subsubfolder) + + + + + + +