Spaces:
Running
on
Zero
Running
on
Zero
File size: 492 Bytes
d59f323 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import numpy as np
from torchvision.transforms.functional import resize, to_pil_image # type: ignore
class DirectResize:
def __init__(self, target_length: int) -> None:
self.target_length = target_length
def apply_image(self, image: np.ndarray) -> np.ndarray:
"""
Expects a numpy array with shape HxWxC in uint8 format.
"""
img = to_pil_image(image, mode='RGB')
return np.array(img.resize((self.target_length, self.target_length)))
|