From 7c530004ba0090545540c307c78d7d451182d677 Mon Sep 17 00:00:00 2001 From: Amy Roberts <22614925+amyeroberts@users.noreply.github.com> Date: Wed, 16 Aug 2023 16:37:07 +0000 Subject: [PATCH 1/2] Update default rescale_factor value --- src/transformers/models/vivit/image_processing_vivit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/vivit/image_processing_vivit.py b/src/transformers/models/vivit/image_processing_vivit.py index 0790c5d82bb5..b93d94d9a722 100644 --- a/src/transformers/models/vivit/image_processing_vivit.py +++ b/src/transformers/models/vivit/image_processing_vivit.py @@ -83,7 +83,7 @@ class VivitImageProcessor(BaseImageProcessor): do_rescale (`bool`, *optional*, defaults to `True`): Whether to rescale the image by the specified scale `rescale_factor`. Can be overridden by the `do_rescale` parameter in the `preprocess` method. - rescale_factor (`int` or `float`, *optional*, defaults to `1/255`): + rescale_factor (`int` or `float`, *optional*, defaults to `1/127.5`): Defines the scale factor to use if rescaling the image. Can be overridden by the `rescale_factor` parameter in the `preprocess` method. offset (`bool`, *optional*, defaults to `True`): @@ -110,7 +110,7 @@ def __init__( do_center_crop: bool = True, crop_size: Dict[str, int] = None, do_rescale: bool = True, - rescale_factor: Union[int, float] = 1 / 255, + rescale_factor: Union[int, float] = 1 / 127.5, offset: bool = True, do_normalize: bool = True, image_mean: Optional[Union[float, List[float]]] = None, From 125119c888130d96694f0613a3345a34c06984e1 Mon Sep 17 00:00:00 2001 From: amyeroberts <22614925+amyeroberts@users.noreply.github.com> Date: Wed, 16 Aug 2023 17:40:11 +0100 Subject: [PATCH 2/2] Formatting --- src/transformers/models/vivit/image_processing_vivit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/vivit/image_processing_vivit.py b/src/transformers/models/vivit/image_processing_vivit.py index b93d94d9a722..af02400e71e6 100644 --- a/src/transformers/models/vivit/image_processing_vivit.py +++ b/src/transformers/models/vivit/image_processing_vivit.py @@ -83,7 +83,7 @@ class VivitImageProcessor(BaseImageProcessor): do_rescale (`bool`, *optional*, defaults to `True`): Whether to rescale the image by the specified scale `rescale_factor`. Can be overridden by the `do_rescale` parameter in the `preprocess` method. - rescale_factor (`int` or `float`, *optional*, defaults to `1/127.5`): + rescale_factor (`int` or `float`, *optional*, defaults to 1/127.5): Defines the scale factor to use if rescaling the image. Can be overridden by the `rescale_factor` parameter in the `preprocess` method. offset (`bool`, *optional*, defaults to `True`):