使用 pytorch 运行预训练模型的框架

PyTorch 简介:

PyTorch 是一个 Python 程序库,我们可以使用 PyTorch 来构建深度学习项目。

PyTorch 的两个特点:

  1. PyTorch 的核心数据结构是张量,张量是一个多维数组,与 NumPy 数组有许多相似之处。
  2. PyTorch 提供了在专用硬件上执行加速数学操作的特性,这使得神经网络结构设计以及在单机或并行计算资源上训练它们变得很方便。

因此,我们可以将 PyTorch 描述为一个在 Python 中为科学计算提供优化支持的高性能库。

PyTorch 大部分是用 C++ 和 CUDA 编写的,CUDA 是一种来自英伟达的类 C++的语言,可以被编译并在 GPU 上以并行方式运行。

使用 pytorch 运行预训练模型的框架

复制代码
import torch
  1. 定义模型类 1.1 自定义模型类 1.2 从 torchvision 模块加载模型: from torchvision import models

  2. 实例化模型类

    resnet101 = models.resnet101()

  3. 给实例化的模型类加载预训练好的参数 3.1 实例化模型类和加载预训练好的权重同时进行(这种情况可以省略第 2 步)

    resnet101 = models.resnet101(pretrained=True) # pretrained=True 指示函数下载 resnet101 在 ImageNet数据集上训练好的权重

3.2 使用模型的 load_state_dict() 方法将预训练权重加载到 resnet101 中

复制代码
model_path = '......'
model_data = torch.load(model_path)
resnet101.load_state_dict(model_data)

3.3 使用 torch.hub 从 github 加载模型(这种情况可以省略第 1、2 步)

复制代码
from torch import hub
resnet101 = hub.load('pytorch/vision:main', 'resnet101', pretrained=True)  # 第一项是 GitHub 存储库的名称和分支,第二项是入口点函数的名称

以上代码将 pytorch/vision 主分支的快照及其权重默认下载到本地的 C:\Users\username.cache\torch\hub 目录下,然后运行 resnet101 入口点函数返回实例化的模型,参数 pretrained=true 会从 ImageNet 获得预训练权重,并加载到 resnet101 中。

  1. 使用 Python 图像操作模块 Pillow 从本地文件系统加载一幅图像

    from PIL import Image # PIL 指的是 pillow
    img = Image.open(".../xxx.jpg")

  2. 使用 TorchVision 模块提供的 transforms 定义一个对输入图像进行预处理的管道

    from torchvision import transforms
    preprocess = transforms.Compose([transforms.Resize(256), # 将输入图像缩放到 256× 256 个像素
    transforms.ToTensor(), # 转换为一个张量
    ])

  3. 使用预处理管道 preprocess 对图像 img 进行预处理

    img_t = preprocess(img)

  4. 给数据添加一个新的维度:批次维度

    batch_t = torch.unsqueeze(img_t, 0)

  5. 进行推理时,我们需要将神经网络置于 eval 模式

    resnet.eval()

  6. eval 模式设置好之后,进行推理

    out = resnet101(batch_t)
    out

......

本文由mdnice多平台发布

相关推荐
昵称为空C4 分钟前
SpringBoot数据存储时区选择,符合国际化和特定时区方案
spring boot·后端
ldj20201 小时前
SpringBoot为什么使用new RuntimeException() 来获取调用栈?
java·spring boot·后端
超龄超能程序猿1 小时前
Spring 应用中 Swagger 2.0 迁移 OpenAPI 3.0 详解:配置、注解与实践
java·spring boot·后端·spring·spring cloud
江南一点雨1 小时前
Tokenizer 和 BPE
后端
风象南1 小时前
SpringBoot配置属性热更新的轻量级实现
java·spring boot·后端
洛阳泰山1 小时前
Spring Boot 整合 Nacos 实战教程:服务注册发现与配置中心详解
java·spring boot·后端·nacos
江南一点雨2 小时前
ChatGPT与最大似然估计
后端
程序员爱钓鱼3 小时前
Go语言实战案例-判断一个数是否为质数
后端·google·go
程序员爱钓鱼3 小时前
Go语言实战案例-读取本地文本文件内容
后端·google·go
mCell11 小时前
Go 并发编程基础:从 Goroutine 到 Worker Pool 实践
后端·性能优化·go