赞
踩
在做迁移学习时,难免会碰到要下载预训练模型参数的时候,但是由于服务器是在国外,在国内下载速度贼慢,这时候怎么办呢?你可能需要这篇博客了!具体步骤如下:
在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文件(这里推荐用第二种,因为这种方式加载的模型更加丰富)。
想要快速下载预训练模型,需要加载一个环境变量
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文件下载地址的分享)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。