zhreshold / mxnet-ssd

MXNet port of SSD: Single Shot MultiBox Object Detector. Reimplementation of https://github.com/weiliu89/caffe/tree/ssd
MIT License
763 stars 339 forks source link

MobileNet-v2 SSD Model #216

Open jacky4323 opened 6 years ago

jacky4323 commented 6 years ago

Hi,

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

Thanks.

zhreshold commented 6 years ago

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

jacky4323 commented 6 years ago

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!

zhreshold commented 6 years ago

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
jacky4323 commented 6 years ago

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

zhreshold commented 6 years ago

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

jacky4323 commented 6 years ago

Hi,

OK, GluonCV is a good work, thanks!