Обучение в Региональной сети предложений только для обнаружения объектов с использованием Tensorflow

Question

Мне нравится тренироваться только RPN network с faster rcnn, используя tensorflow.В caffe я обучал только RPN.

В тензорном потоке это руководство показывает обучение настроенной сети.Затем другой учебник здесь обсуждал, как тренироваться, используя модели из modelzoo .

Так что для моего случая, чтобы обучать только части RPN, какой подход мне подходит?Скажите, если я следую первому подходу, определяющему сеть CNN, как я могу загрузить pretrained model для transfer learning.

Answer 1

Чтобы обучить сеть RPN только с использованием API обнаружения объектов tenorflow, вам просто нужно изменить конфигурационные файлы, потому что API поддерживает опцию только RPN, как описано здесь .Чтобы получить модель только для RPN, просто добавьте опцию 'number_of_stages : 1' в конфигурации модели area (аналогично 'num_of_classes : 90').(Вы также можете удалить некоторые параметры второго этапа из файла конфигурации, поскольку они бесполезны, если для number_of_stages установлено значение 1).

Что касается второго вопроса о трансферном обучении, он такой же, как и в случае обученияfaster rcnn сеть, просто нужно убедиться, что from_detection_checkpoint верно и finetune_checkpoint_path правильно в файле конфигурации .

В процессе обучения вы можете использовать тензорную доску для проверки состоянияПосле выполнения оценки на вкладке «Изображения» можно увидеть, что предложения по регионам визуализируются на изображении, а все метки обозначены как «объект + показатель доверия».