Поезд Mobilnet V2 с трансферным обучением в Tensorflow

1

Folks,

Как я могу сделать передачу/изучение домена с помощью mobilenet v2? Я использую код обучения mobilnet v1 для обучения mobilenet v1, но есть ли написанный код для обучения v2 либо в репозитории моделей, либо в репозитории tensorflow?

https://github.com/tensorflow/models/tree/master/research

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/image_retraining

Теги:
tensorflow
training-data
models

2 ответа

0
Лучший ответ

@Asher Mancinelli, на самом деле, этот код делает именно то, что мне нужно: https://raw.githubusercontent.com/tensorflow/hub/r0.1/examples/image_retraining/retrain.py

Поделиться
1

У этого есть модули mobilenet v2 tfslim, а также файлы контрольных точек для восстановления весов, уже прошедших обучение людей с тензонами. Вы можете использовать этот пример для перепрограммирования модели, или вы можете использовать более простой подход к этому учебнику. Если вы ищете переносимость, возможно, захотите взглянуть на TFLite, что позволит вам взять любую модель тензорного потока и легко оптимизировать ее для мобильных устройств с помощью инструментов построения на голове tensorflow: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools.

Удачи

Поделиться

Ещё вопросы

Сообщество Overcoder
Наверх
Меню