dmlc / gluon-cv

Gluon CV Toolkit
http://gluon-cv.mxnet.io
Apache License 2.0
5.82k stars 1.21k forks source link

[Feature Request] attach some symbol file for model zoo(details of model connection) #198

Open jacky4323 opened 6 years ago

jacky4323 commented 6 years ago

Hi,

To see the connection of pretrained network,it can easly use visualize method,like the code below. However,I think symbol file is more clearly than visualize the network (similar to caffe prototext),because we probably missing some details of the network connection or some parameters. Is there any plan to attach some symbol file ? also , it can claims that the model is the same as the paper thanks!

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

image

zhreshold commented 6 years ago

Probably will add in the next release.

jacky4323 commented 6 years ago

Hi @zhreshold Does it have some darknet pretrained model? (not only yolov3 due to its complexity) Could you please add some lightweight pretrained model in this direction?

I think some model has benefited at lightweight(like mobilenet、shufflenet、also some lightweight modified vgg etc...), Although these kinds of model maybe have lower accuracy, they can real-time run on the device(30fps). It is attractive to some people whose application at embedded system

thanks!