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