当前位置: 首页 > news >正文

每天五分钟深度学习框架pytorch:视觉工具包torchvison

本文重点

在pytorch深度学习框架中,torchvision是一个非常优秀的视觉工具包,我们可以使用它加载一些著名的数据集,然后我们可以使用它来加载网络模型,比如vgg,resnet等等,还可以使用它来预处理一些图片数据,本节课程我们将学习一下它的使用方式。

torchvision的四部分

torchvision.datasets:流行视觉数据集的数据加载器

torchvision.models:流行的模型体系结构的定义,例如AlexNet,VGG和ResNet以及预先训练的模型。

torchvision.transforms:常见的图像预处理,例如随机裁剪,旋转等。

torchvision.utils:保存张量(3 x H x W)作为图像到磁盘,给定一个小批量创建图像网格等。

torchvision.datasets

torchvision.datasets的父类是torhc.utils.data.Dataset,封装了很多常用的数据集,那么我们可以使用它来读取一些常用的数据集,比如

相关文章:

  • 什么是直播美颜SDK?跨平台安卓、iOS美颜SDK开发实战详解
  • 【递归,搜索与回溯算法篇】专题(一) - 递归
  • Python爬虫(22)Python爬虫进阶:Scrapy框架动态页面爬取与高效数据管道设计
  • 【官方题解】StarryCoding 入门教育赛 2 | acm | 蓝桥杯 | 新手入门
  • NLP基础
  • Java 23种设计模式 - 结构型模式7种
  • c++:迭代器(Iterator)
  • git相关
  • 今日行情明日机会——20250509
  • 从设计到开发,原型标注图全流程标准化
  • 深度学习 ———— 迁移学习
  • 自动驾驶的“眼睛”:用Python构建智能障碍物检测系统
  • 2025医疗信息化趋势:健康管理系统如何重构智慧医院生态
  • 【新品发布】VXI可重构信号处理系统模块系列
  • 内网渗透——红日靶场三
  • C++ stl中的list的相关函数用法
  • 线程邮箱框架与示例
  • 星云智控:物联网时代的设备守护者——卓伊凡详解物联网监控革命-优雅草卓伊凡
  • 【FreeRTOS】1、学习FreeRTOS创建任务
  • Jenkins Maven 带权限 搭建方案2025
  • 沃旭能源因成本上升放弃英国海上风电项目,或损失近40亿元
  • 讲座预告|全球贸易不确定情况下企业创新生态构建
  • 复旦发文缅怀文科杰出教授裘锡圭:曾提出治学需具备三种精神
  • 国家卫健委:有条件的二级及以上综合医院要开设老年医学科
  • 商务部再回应中美经贸高层会谈
  • 上海优化营商环境再攻坚,企业和机构有哪些切实感受?