pytorch零基础实现手写数学符号识别项目(二)——数据加载以及模型浅试

数据加载以及模型浅试

  • 前言
  • 数据加载
    • 加载单个数据
    • 数据加载器
      • 参数
      • 过程解析
    • 批量数据加载
  • 模型
    • ResNet模型
      • 模型结构
      • 模型修改
  • 结尾

前言

在上一篇文章中,我们着重分析了数学符号数据集的特征,并且做好了真实标签与数字标签的映射,同时为了防止内存溢出,我们将图像的路径和类别存放到了txt文件中实行随用随取。本篇文章我们将在上一篇文的基础上定义数据的加载方式以及在resnet下进行模型的训练。

数据加载

加载单个数据

torchvision中,可以使用io.read_image打开适用于torch的图片,还记得上篇我们提到在加载图片时需要使用transforms.ToTensor将图片转换成适用于torch

你可能感兴趣的:(零基础上手计算机视觉项目,python,pytorch,深度学习,计算机视觉,人工智能)