From 53774bb7da6b3ac048102623737a463f6cf3e57f Mon Sep 17 00:00:00 2001 From: JianHe0628 <jl02958@surrey.ac.uk> Date: Thu, 5 Dec 2024 16:43:02 +0000 Subject: [PATCH] changing for video --- Extract_HaMeR_Single.py | 10 +++++----- Inject_Json.py | 14 +++++++------- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py index 5cc9e7c..ff300ca 100644 --- a/Extract_HaMeR_Single.py +++ b/Extract_HaMeR_Single.py @@ -100,11 +100,11 @@ def main(args, model, renderer, device): # Get all demo images ends with .jpg or .png img_paths = [img for end in args.file_type for img in image_folder.glob(end)] - # #MeinDGS Specific - # img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame'))) + #VideoFile Specific + img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame'))) - #For Phoenix - img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removeprefix('images').removesuffix('.png'))) + # #For Phoenix + # img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removeprefix('images').removesuffix('.png'))) img_cv2 = cv2.imread(str(img_paths[0])) if args.bbox == 'True': @@ -282,7 +282,7 @@ if __name__ == '__main__': args = parser.parse_args() - args.bbox = 'False' + args.bbox = 'True' args.out_folder = os.path.join(args.out_folder, os.path.basename(args.img_folder)) print(f"Output folder: {args.out_folder}") diff --git a/Inject_Json.py b/Inject_Json.py index eccd598..86b00bb 100644 --- a/Inject_Json.py +++ b/Inject_Json.py @@ -136,7 +136,9 @@ def main(args, model, device): raise ValueError("Input is not a lmdb file") img_paths = [img for end in args.file_type for img in image_folder.glob(end)] - img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame'))) + # img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame'))) + #For Phoenix + img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removeprefix('images').removesuffix('.png'))) assert len(img_paths) == len(hand_pose_list), f"Number of images {len(img_paths)} and lmdb files {len(hand_pose_list)} do not match" start_frame, end_frame = map(int, args.frame_range.split(',')) @@ -347,13 +349,11 @@ if __name__ == '__main__': args = parser.parse_args() - FileName = '1176549/1176549_1a1' - args.vid = f'/vol/vssp/datasets/mixedmode/mein-dgs-korpus/RawData/{FileName}.mp4' - args.lmdb_input = f'/mnt/fast/nobackup/scratch4weeks/jl02958/MeinDGS_HaMeR/{FileName}.lmdb' - args.out_folder = f'/vol/research/signVision/Projects/BSLboundaries/Test/{FileName.split("/")[-1]}' + Filename = '31May_2011_Tuesday_heute-1900' + args.img_folder = f'/vol/vssp/datasets/singlevideo/phoenix-2014T/features/fullFrame-210x260px/train/{Filename}' + args.lmdb_input = f'/mnt/fast/nobackup/scratch4weeks/jl02958/Phoenix2014T_HaMeR/train/{Filename}/{Filename}.lmdb' + args.out_folder = f'/vol/research/signVision/Projects/BSLboundaries/Test/{Filename}' - args.frame_range = '650,850' - args.custom_bbox = '100,0,530,360' args.bbox = False args.save_mesh = False -- GitLab