当前位置:   article > 正文

一招解决mxnet预训练模型参数下载慢/git clone慢的问题_mxnet模型载入慢

mxnet模型载入慢

      在做迁移学习时,难免会碰到要下载预训练模型参数的时候,但是由于服务器是在国外,在国内下载速度贼慢,这时候怎么办呢?你可能需要这篇博客了!具体步骤如下:

  • 找到你想要下载的预训练模型

mxnet里面,通过model_zoo来实现预训练模型加载,mxnet提供了两种方式

  1. ####第一种方式
  2. from mxnet.gluon.model_zoo import vision as models
  3. models.get_model("预训练模型名字",pretrained=True)
  4. ####第二种方式
  5. from gluoncv import model_zoo
  6. model_zoo.get_model("预训练模型名字",pretrained=True)

在上面两种方式中选择一种保存成py文件(这里推荐用第二种,因为这种方式加载的模型更加丰富)。

  • 编写下载shell脚本

想要快速下载预训练模型,需要加载一个环境变量

export MXNET_GLUON_REPO=https://apache-mxnet.s3.cn-north-1.amazonaws.com.cn/

然后在脚本中运行上面的py文件。

整个shell脚本如下:

  1. export MXNET_GLUON_REPO=https://apache-mxnet.s3.cn-north-1.amazonaws.com.cn/
  2. python3 xx.py (根据自己的情况进行修改)

补:

》Mac下git命令太慢,可以考虑将链接复制到码云上,然后再从其上下载就可以快速得到你想要的资源了。

Keras导入预训练模型 from keras.applications.vgg19 import VGG19等

  TensorFlow 导入预训练模型from tensorflow.keras.applications.inception_v3 import InceptionV3等

   Torch 导入预训练模型 from torchvision.models import resnet18等

》快速下载yolov3预训练权重的方法

  1. ###详情可参考https://blog.csdn.net/Thomson617/article/details/99972062这篇博客
  2. https://pjreddie.com/media/files/yolov3.weights ##直接用迅雷等工具下
  3. https://pjreddie.com/media/files/yolov3-spp.weights ##直接下载yolov3-spp的权重

 

参考链接:https://www.jianshu.com/p/a9e1fd5dff68

                  https://blog.csdn.net/Thomson617/article/details/99972062(关于yolov3.weights文件下载地址的分享)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Li_阴宅/article/detail/894317
推荐阅读
相关标签
  

闽ICP备14008679号