From a9ff482026d338545d3ce15be82d772df8dc42d0 Mon Sep 17 00:00:00 2001
From: JianHe0628 <jl02958@surrey.ac.uk>
Date: Wed, 4 Dec 2024 12:43:14 +0000
Subject: [PATCH] changing sorting to Phoenix specific
---
Extract_HaMeR_Single.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py
index 3b11b8e..d6dc866 100644
--- a/Extract_HaMeR_Single.py
+++ b/Extract_HaMeR_Single.py
@@ -98,7 +98,12 @@ def main(args, model, renderer, device):
image_folder = Path(args.img_folder)
# 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)]
- img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame')))
+
+ # #MeinDGS 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')))
img_cv2 = cv2.imread(str(img_paths[0]))
if args.bbox == 'True':
--
GitLab