tensorflow / models

Models and examples built with TensorFlow
Other
77.16k stars 45.76k forks source link

object detection: using more feature extractors with faster RCNN (feature request) #1924

Closed gdelab closed 7 years ago

gdelab commented 7 years ago

System information

Describe the problem

I'm trying to perform object detection on a custom, relatively easy dataset (with ~30k samples). I've successfully used Faster_RCNN with Resnet101_v1 (final mAP 0.9) and inception_resnet_v2 feature extractors (training in progress). Now I would like my model to run faster but still keep good performance, so I'd like to compare the ones I have, with SSD running with various versions of mobile_net. However, to know which changes in performance come from SSD and which come from the feature extractor, I'd like to also try Faster-RCNN with mobile_nets. It's also possible that this yields the tradeoff I need between performance and inference time (faster RCNN being good and slow, and mobile_nets fast).

How can I make mobile_nets compatible with Faster-RCNN?

michaelisard commented 7 years ago

This question is better asked on StackOverflow since it is not a bug or feature request. There is also a larger community that reads questions there. Thanks!

akashjobanputra commented 7 years ago

@gdelab if you have posted the query on StackOverflow, can you share the link to your post ? Thanks!