mxnet-ssd icon indicating copy to clipboard operation
mxnet-ssd copied to clipboard

MobileNet-v2 SSD Model

Open jacky4323 opened this issue 7 years ago • 6 comments

Hi,

Did you have training MobileNet-v2 SSD Model? Are there some pretrained model I can use?

Thanks.

jacky4323 avatar Jul 05 '18 13:07 jacky4323

Please follow up GluonCV for the pre-trained models. Mobilenet v2 SSD is missing, but you can expect more models coming.

zhreshold avatar Jul 05 '18 17:07 zhreshold

Hi,, I only see some of the parameter and Jason file Dose it have some file like symbol which can see the model connection and filter numbers more clearly?

Many Thanks!

jacky4323 avatar Jul 06 '18 08:07 jacky4323

given a gluon net, you can convert it to symbol very quickly:

import gluoncv as gcv
import mxnet as mx
net = gcv.model_zoo.get_model('ssd_512_mobilenet1.0_voc', pretrained=True)
sym = net(mx.sym.var('data'))
# visualize it like usual

zhreshold avatar Jul 06 '18 17:07 zhreshold

Hi, I follow your steps, and visualize the model network, mx.viz.plot_network(symbol=sym) to plot the network ,thanks. I think symbol/mobilenet.py is more clear than visualize the network (similar to caffe prototext) Is there any plan to attach some symbol? also , it can claims that the model is the same as the paper thanks!

image

jacky4323 avatar Jul 08 '18 07:07 jacky4323

You can send a feature request to GluonCV, we can take it into consideration in the roadmap!

zhreshold avatar Jul 08 '18 21:07 zhreshold

Hi,

OK, GluonCV is a good work, thanks!

jacky4323 avatar Jul 08 '18 23:07 jacky4323