diff --git a/Extract_HaMeR_Single.py b/Extract_HaMeR_Single.py index 1481e4a48a3999dd1523615f558a4ec1dcf854ae..5cc9e7c8590293f4289d3838c0e7e35ca7571e49 100644 --- a/Extract_HaMeR_Single.py +++ b/Extract_HaMeR_Single.py @@ -96,6 +96,7 @@ def main(args, model, renderer, device): print(f"Total Time for Video to Image: {(time.time()-start_time)} Seconds") else: image_folder = Path(args.img_folder) + fps = None # 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)]