определить input_size самостоятельно, когда я применяю обучение переносу в моих собственных наборах данных

Question

Я использую тензорный поток для глубокого обучения.

Я хочу попробовать перенести обучение в мои собственные наборы данных, и я скачал модель inceptionv3 с веб-сайта tenorflow. Я также нахожу демо, но я нахожу модель input_size 299 * 299 * 3. Я хочу определить input_size самостоятельно. Потому что начальная модель Keras v3 может определить input_size самостоятельно. Например, input_size - 512 * 512 * 3.

Я не использую функцию изменения размера.

Я пытался сделать следующее: введите описание изображения здесь

но я получил следующую ошибку: введите описание изображения здесь

Когда я изменяю его на 299 * 299 * 3, код работает нормально.

Answer 1

Что касается InceptionV3, вы можете использовать любой размер изображения, и предварительная обработка tenorflow будет принимать размер изображения. Официальный начальный модуль Tensorflow включает https://github.com/tensorflow/models/blob/master/research/inception/inception/image_processing.py, в котором вы можете указать размер изображения, которое вы хотите использовать. Затем модель может быть переобучена с использованием этого нового размера.

Answer 2

Вы не можете легко изменить размер ввода обучаемой модели. Веса обученной модели знают только, как обрабатывать ввод такого размера. Если вы хотите использовать предварительно обученные веса, лучше всего изменить размеры изображений до ожидаемого размера.