Search code examples
tensorflowobject-detection-api

How to add feature extractor netwrok for example mobilenetv2 to tensorflow's object detection API


This tutorial discusses how to use objection detection API at tensorflow. I am looking for the tutorial explaining how to add feature extractor such as mobilenetV2 to tensorflow's object detection framework.


Solution

  • Have you checked out the Tensorflow provided Model Zoo? :)

    It includes various object detection models with various feature extractors such as MobileNet, Inception, ResNet etc.

    Here below you will find a link to the Tensorflow Detection Model Zoo, where you can choose detection model architectures, Region-Based (R-CNN) or Single Shot Detector (SSD) models, and feature extractors.

    https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

    You can download a frozen graph of the pre-trained models based on COCO, Kitti and Open-Images etc.