diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py index d6dc866d7e754dccbcf0342a3676451dafb135a4..1481e4a48a3999dd1523615f558a4ec1dcf854ae 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':