Я обучил модель vgg_ unet в своем собственном наборе данных и сохранил модель как model.h5. После сохранения модели я загружаю модель, используя from keras.models import load_model. Теперь, когда я пытаюсь предсказать и сохранить результат как изображение, используя приведенный ниже код, он выдает мне сообщение об ошибке: OpenCV(4.1.2) /io/opencv/modules/imgcodecs/src/loadsave.cpp:668: error: (-215:Assertion failed) image.channels() == 1 || image.channels() == 3 || image.channels() == 4 in function 'imwrite_'. Я хочу вывод как изображение. Форма выходного массива: (1, 102400, 256). Пожалуйста, предложите мне что-нибудь о том, как сохранить этот массив numpy как изображение.
Код для прогноза приведен ниже:
inp="drive/My Drive/CT Scan Segmentation/dataset/test_img/test9.jpeg"
img = cv2.imread(inp)
img_pred = image.load_img('drive/My Drive/CT Scan Segmentation/dataset/test_img/test9.jpeg', target_size = (640, 640))
img_pred = image.img_to_array(img_pred)
img_pred = np.expand_dims(img_pred, axis = 0)
out = model.predict(img_pred)
cv2.imwrite("drive/My Drive/CT Scan Segmentation/dataset/test_mask/test9.png", out)