переподготовка модели, когда меня интересует только подкатегория существующих классов - PullRequest
0 голосов
/ 09 июня 2019

Я хочу использовать обученную модель из API-интерфейса обнаружения объектов тензорного потока, в частности, я хочу использовать quick_rcnn_inception_resnet_v2_atrous_oid_v4, обученную в Google Open Imaged. Я не заинтересован в обнаружении всех 601 классов, а скорее хотел бы обнаружить 10 подклассов. Получу ли я повышение точности, если сохраню последний слой, или лучше отфильтровать слои, которые мне не интересны, после того, как модель будет сделана с предсказанием. Если я пошел с сохранением, это нормально, чтобы сохранить модель с изображениями из Google открытых изображений снова или лучше использовать другие данные.

1 Ответ

0 голосов
/ 13 июня 2019

Этот официальный пример , кажется, поможет вам.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...