24 lines
782 B
Python
24 lines
782 B
Python
import cv2 as cv
|
||
import numpy as np
|
||
|
||
from main import img_size as size
|
||
|
||
def image_transform(image: np.ndarray) -> np.ndarray:
|
||
'''
|
||
Трансформирует изображение нужным образом.
|
||
@param image: Исходная матрица с представлением изображения.
|
||
'''
|
||
image = cv.resize(image, (size[0], size[1]))
|
||
return image[:, :, ::-1]
|
||
|
||
|
||
def get_image_as_array(image_name: str) -> np.ndarray:
|
||
'''
|
||
Получает изображение из файла и нормализует его.
|
||
@param image_name: Путь до изображения.
|
||
'''
|
||
image = cv.imread(image_name)
|
||
image: np.ndarray # приведение типов
|
||
image = image_transform(image)
|
||
return image
|