Распознавание изображений с помощью TensorFlow

Я новичок в TensorFlow, и я ищу помощь по распознаванию изображений. Есть ли пример, демонстрирующий, как использовать TensorFlow для обучения ваших собственных цифровых изображений для распознавания изображений, таких как модель image-net, используемая в учебнике по распознаванию образов TensorFlow.

Я посмотрел на обучение модели CIFAR-10, но, похоже, это не дает примеров для обучения ваших собственных изображений.

Ответ 1

Я бы порекомендовал использовать обучаемую начальную модель Google для распознавания образов. Пожалуйста, обратитесь к примеру "Как перенести начальный финальный слой для новых категорий" на веб-сайте tendorflow. Он находится в https://www.tensorflow.org/versions/r0.9/how_tos/image_retraining/index.html.

Использование обученной модели легко и может обеспечить разумную точность. Вы просто просто загружаете модель своим собственным набором данных. Последний класс классификации начала Google будет изменен, и мы будем тренировать только последний слой. Для нескольких тысяч изображений в нескольких категориях требуется только несколько часов, чтобы закончить обучение. Обратите внимание: для того, чтобы использовать этот пример, вам нужно построить shadoworflow из источника.

Я использую функцию обучения передаче и получаю очень хорошие результаты. Чтобы проиллюстрировать преимущества передачи обучения, я сравниваю "Передача обучения в обученном GoogleNet" с помощью "Построить и обучить 5-слойный Convnet с нуля". Задача классификации выполняется на 5000 изображений с 5 категориями.

Создайте 5-слойный сверток и обучите его с нуля

Использовать обучение передачи в обученном GoogleNet

См. этот простой пример: https://www.youtube.com/watch?v=QfNvhPx5Px8 (создайте классификатор изображений TensorFlow за 5 минут)

Ответ 2

Это репо дает предварительную переподготовку изображений для ваших собственных изображений.

Добавьте свои изображения в папку набора данных training_

Добавить все больше и больше изображений (по крайней мере, 30 изображений для каждого класса.)

Сначала выполните bash script, чтобы загрузить файлы Imagenet.

Затем запустите python script, чтобы перенастроить ваши изображения, которые выдают точность проверки и т.д.,

https://github.com/shivakrishna9/tensorflow-retrain