API обнаружения объектов Tensorflow только для классификации - PullRequest
0 голосов
/ 05 марта 2020

Я знаю, как точно настроить для обнаружения одну из предварительно обученных моделей COCO, доступных в модели обнаружения TensorFlow zoo git repo. Любопытство вспыхнуло в тот момент, когда я заметил параметр " fine_tune_checkpoint_type " в файле pipe.config, поставляемом с контрольной точкой. Я думал, что смогу использовать модель только для классификации, предоставив в качестве ограничивающего прямоугольника весь размер изображения ... но похоже, что я слишком переоснащаю. Таким образом, вопрос в том, можно ли просто заставить модель выполнить классификацию ? (Мне не нужны ограничительные рамки)

Для справки я хотел бы использовать " ssd_mobilenet_v2_coco ".

Заранее спасибо

...