在做训练的时候需要调用预训练好的模型,模型存储在远程服务器上,需要在运行前下载到本地然后调用。需要使用到urllib模块,如果未安装,则需要在安装环境中使用如下命令安装
pip install urllib
调用下载模块:
import os
import urllib
import sys
#远程下载链接
url='http://hello/download.zip'
#下载到本地的路径和保存的文件名
#注意此处的路径不能是不存在的路径,若想创建新路径,使用os.mkdir('路径')
saveFile = './download.zip'
#下载模块
urllib.request.urlretrieve(url,saveFile)
#解压路径,注意解压的时候,源文件内有没有二级的文件,自己容易文件路径想当然,调用路径出错
upZipPath = './upzip'
#若解压路径不存在,则创建解压路径
if not os.path.isdir(upZipPath):
os.mkdir(upZipPath)
#开始解压
#判断是否是压缩文件
if zipfile.is_zipfile(fileName):
#读取压缩文件
fz = zipfile.ZipFile(fileName, 'r')
#提取文件到解压目录
for file in fz.namelist():
fz.extract(file, pretrained_dir)
else:
#否则返回失败
print('Cannot load model from', remote_name)
return None