search

tensorflow / models

Models and examples built with TensorFlow
Other
76.95k stars 45.79k forks source link

Faster rcnn mobilene_v2 and mobilenet_v3 support #8793

Open anshkumar opened 4 years ago

anshkumar commented 4 years ago

Since faster_rcnn_mobilenet_v1_feature_extractor is already provided, adding mobilenet_v2 and mobilenet_v3 should be easy. Mobilenet_v2 and v3 provides better accuracy than v1.

I've written feature_extractor for both faster_rcnn_mobilenet_v2_feature_extractor.py and faster_rcnn_mobilenet_v3_feature_extractor.py. It can be added to the official repository.

pkulzc commented 4 years ago

Could you create a pull request with your change? We can review it.

anshkumar commented 4 years ago

Created one.