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