Как проверить свои собственные данные (одно изображение), используя уже обученную модель?

Question

Я пытаюсь запустить этот репозиторий . И, пытаясь проверить с моими собственными данными. В то время как я пытался это сделать, я получаю эту ошибку:

Traceback (последний вызов был последним): файл "test.py", строка 34, в выходных данных = прогноз (модель, входные данные) Файл "/home/kbdp5524/Downloads/DenseDepth-master/utils.py", строка 12, в предсказываемых предсказаниях = файл model.predict (images, batch_size = batch_size) "/ home / kbdp5524 / anaconda3 / envs / densitydepth / lib / python3 .7 / site-packages / keras / engine / training.py ", строка 1441, в файле прогнозирования x, _, _ = self._standardize_user_data (x) File" / home / kbdp5524 / anaconda3 / envs / densdepth / lib /python3.7/site-packages/keras/engine/training.py ", строка 579, в _standardize_user_data exception_prefix = 'input') Файл" / home / kbdp5524 / anaconda3 / envs / densitydepth / lib / python3. 7 / site-packages / keras / engine / training_utils.py ", строка 145, в строке standardize_input_data (data_shape))

ValueError: Ошибка при проверке ввода: ожидается, что input_1 будет иметь форму (нет, нет, 3) но получил массив с формой (480, 640, 4)

Это поезд test.py и . py код

Может кто-нибудь, пожалуйста, помогите мне через это. Я новичок в кодировании:)

Answer 1

Ваше входное изображение имеет форму (Нет, Нет, 4) изображения RGBA, но модель требует, чтобы 3-канальное изображение означало (Нет, Нет, 3) RGB. Таким образом, вы должны конвертировать ваше изображение в 3 канала. Вы можете сделать это, используя функцию OpenCV cv2.cvtColor(Image, cv2.COLOR_RGBA2RGB ).

Чтобы установить openCV: pip install python -opencv