Как сделать перевод обучения без потери ящиков обнаружения - PullRequest
0 голосов
/ 21 октября 2019

Я делаю переучивание на детекторе объектов ImagenetV2. Когда я использую предварительно обученную модель, model.predict () возвращает результат, который содержит координаты блоков обнаружения объекта. После обучения передачи я получаю только один выходной номер, от нуля до одного. ниже приведен мой псевдокод.

flatten last layer of pretrained model
add layer with 128 nodes
add layer with 1 node, activation sigmoid

Я хочу знать, как выполнить переносное обучение, чтобы моя модель нашла объект на изображении и вернула координаты ограничивающего прямоугольника с доверием

...