Skip to content

Commit 7693600

Browse files
committed
Update rescale method to reflect changes in huggingface#18677
1 parent 6ec9dbb commit 7693600

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/transformers/image_transforms.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def to_channel_dimension_format(image: np.ndarray, channel_dim: Union[ChannelDim
6969
raise ValueError("Unsupported channel dimension format: {}".format(channel_dim))
7070

7171

72-
def rescale(
72+
def rescale_image(
7373
image: np.ndarray, scale: Union[float, int] = 255, data_format: Optional[ChannelDimension] = None, dtype=np.float32
7474
) -> np.ndarray:
7575
"""
@@ -127,7 +127,7 @@ def to_pil_image(
127127
# PIL.Image can only store uint8 values, so we rescale the image to be between 0 and 255 if needed.
128128
do_rescale = isinstance(image.flat[0], float) if do_rescale is None else do_rescale
129129
if do_rescale:
130-
image = rescale(image, 255)
130+
image = rescale_image(image, 255)
131131
image = image.astype(np.uint8)
132132
return PIL.Image.fromarray(image)
133133

src/transformers/models/glpn/image_processing_glpn.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from transformers.utils.generic import TensorType
2323

2424
from ...image_processing_utils import BaseImageProcessor, BatchFeature
25-
from ...image_transforms import rescale, resize, to_channel_dimension_format
25+
from ...image_transforms import rescale_image, resize, to_channel_dimension_format
2626
from ...image_utils import ChannelDimension, get_image_size, is_batched, to_numpy_array, valid_images
2727
from ...utils import logging
2828

@@ -93,7 +93,7 @@ def resize(
9393
image = resize(image, (new_h, new_w), resample=resample, data_format=data_format, **kwargs)
9494
return image
9595

96-
def rescale(
96+
def rescale_image(
9797
self, image: np.ndarray, scale: Union[int, float], data_format: Optional[ChannelDimension] = None, **kwargs
9898
) -> np.ndarray:
9999
"""
@@ -110,7 +110,7 @@ def rescale(
110110
- `ChannelDimension.FIRST`: image in (num_channels, height, width) format.
111111
- `ChannelDimension.LAST`: image in (height, width, num_channels) format.
112112
"""
113-
return rescale(image=image, scale=scale, data_format=data_format, **kwargs)
113+
return rescale_image(image=image, scale=scale, data_format=data_format, **kwargs)
114114

115115
def preprocess(
116116
self,
@@ -172,7 +172,7 @@ def preprocess(
172172
images = [self.resize(image, size_divisor=size_divisor, resample=resample) for image in images]
173173

174174
if do_rescale:
175-
images = [self.rescale(image, scale=1 / 255) for image in images]
175+
images = [self.rescale_image(image, scale=1 / 255) for image in images]
176176

177177
images = [to_channel_dimension_format(image, data_format) for image in images]
178178

0 commit comments

Comments
 (0)