diff --git a/Extract_HaMeR_Jsons.py b/Extract_HaMeR_Jsons.py index eb2aa93fc8c387ac3181ec88a7d711e0f351cac4..cbaba55f9d0b9cc087015a14dd454cd2ef589a9e 100644 --- a/Extract_HaMeR_Jsons.py +++ b/Extract_HaMeR_Jsons.py @@ -313,9 +313,11 @@ if __name__ == '__main__': # keypoint detector cpm = ViTPoseModel(device) - args.in_folder = '/vol/research/signVision/Projects/Hamer_Feature_Test/Glosses/test' + # args.in_folder = '/vol/research/SignMotion/2024.02.20_Capture_1/glosses' Folder_List = [os.path.join(args.in_folder, f) for f in os.listdir(args.in_folder)] - args.out_folder = '/vol/research/signVision/Projects/Hamer_Feature_Test/Batch_Test' + Rachel_Path = '/vol/research/SignFeaturePool/Rachel_Gloss_Features/HaMeR_Features' + + args.out_folder = os.path.join(Rachel_Path, os.path.basename(args.in_folder.removesuffix('/glosses'))) if Folder_List[0].endswith('.mp4'): for vid_file in Folder_List: