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

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

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

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

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

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

  • 编写下载shell脚本

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

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

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

整个shell脚本如下:

export MXNET_GLUON_REPO=https://apache-mxnet.s3.cn-north-1.amazonaws.com.cn/
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预训练权重的方法

###详情可参考https://blog.csdn.net/Thomson617/article/details/99972062这篇博客
https://pjreddie.com/media/files/yolov3.weights  ##直接用迅雷等工具下
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文件下载地址的分享)

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值