From 14770d980b5ccf9fdc9c095a2c9b42a4dd0de312 Mon Sep 17 00:00:00 2001
From: JianHe0628 <jl02958@surrey.ac.uk>
Date: Wed, 4 Dec 2024 12:54:57 +0000
Subject: [PATCH] fix sort
---
Extract_HaMeR_Single.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py
index d6dc866..1481e4a 100644
--- a/Extract_HaMeR_Single.py
+++ b/Extract_HaMeR_Single.py
@@ -103,7 +103,7 @@ def main(args, model, renderer, device):
# 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_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':
--
GitLab