import cv2 as cv import numpy as np from main import img_size as size def image_transform(image): image = cv.resize(image, (size[0], size[1])) return image[:, :, ::-1] def get_image_as_array(image_name): image = cv.imread(image_name) image: np.ndarray # приведение типов image = image_transform(image) return image