MXNet中有很多训练好的模型,直接在网页上下载一直报错,后来在https://github.com/apache/incubator-mxnet/blob/master/example/image-classification/common/modelzoo.py中找到一个下载代码。修改以后可直接调用用于下载。
import os
import subprocess
import os
import errno
import mxnet as mx
_base_model_url = 'http://data.mxnet.io/models/'
_default_model_info = {
'imagenet1k-vgg-16': {'symbol':_base_model_url+'imagenet/vgg/vgg16-symbol.json',
'params':_base_model_url+'imagenet/vgg/vgg16-0000.params'},
'imagenet1k-inception-bn': {'symbol':_base_model_url+'imagenet/inception-bn/Inception-BN-symbol.json',
'params':_base_model_url+'imagenet/inception-bn/Inception-BN-0126.params'},
'imagenet1k-resnet-18': {'symbol':_base_model_url+'imagenet/resnet/18-layers/resnet-18-symbol.json',
'params':_base_model_url+'imagenet/resnet/18-layers/resnet-18-0000.params'},
'imagenet1k-resnet-34': {'symbol':_base_model_url+'imagenet/resnet/34-layers/resnet-34-symbol.json',
'params':_base_model_url+