AIY проект классификации изображений на Ubuntu

Question

(venv) nvidia @ tegra-ubuntu: ~ / tenorflow-for-poets-2 $ IMAGE_SIZE = 128 (venv) nvidia @ tegra-ubuntu: ~ /ensorflow-for-poets-2 $ ARCHITECTURE = 'mobilenet_0.25_128_quantized'(venv) nvidia @ tegra-ubuntu: ~ / tenorsflow-for-poets-2 $ ARCHITECTURE = "mobilenet_0.25_128_quantized"

(venv) nvidia @ tegra-ubuntu: ~ / тензор потока для поэтов-2 $ python -m scripts.retrain \

Я использовал приведенный выше код для обучения мобильной сети на Ubuntu, и обучение работает нормально .... Однако, когда я пытаюсь протестировать новые изображения, я использовал приведенный ниже код.

(venv) nvidia @ tegra-ubuntu: ~ / тензор потока для поэтов-2 $ python -m scripts.label_image \

--graph=tf_files/retrained_graph.pb  \
--image=tf_files/Test_Images_Randomized/2018-10-23_14-24-57_696.jpg

я получилэта ошибка.

будет очень признателен за помощь в решении этой проблемы.Я застрял на этом с прошлой недели .... Заранее спасибо

ValueError: Невозможно передать значение формы (1, 224, 224, 3) для Tensor 'import / input: 0', который имеетshape '(1, 128, 128, 3)'

Answer 1

Обязательно настройте разрешение входного изображения в сценарии 'label_image'.

Возможно, оно будет настроено на использование модели с полным разрешением по умолчанию с размером входного файла 224x224.Это должно исправить вашу ошибку измерения.:)

Исправить это:

def read_tensor_from_image_file(file_name,
                            input_height=224,
                            input_width=224,
                            input_mean=0,
                            input_std=255):

К

def read_tensor_from_image_file(file_name,
                            input_height=128,
                            input_width=128,
                            input_mean=0,
                            input_std=255):