datasets模块:
提供一些数据集
数据加载
ImageFolder
DatasetFolder
models模块:
经典网络架构的实现 (可以拿到网络架构,训练权重可用于迁移学习,偏置参数)
例:基于pytorch导入vgg16代码
import torchvision.models as models
vgg16=model.vgg16()
transforms模块
用于训练数据的图像预处理
datasets模块:
提供一些数据集
数据加载
ImageFolder
DatasetFolder
models模块:
经典网络架构的实现 (可以拿到网络架构,训练权重可用于迁移学习,偏置参数)
例:基于pytorch导入vgg16代码
import torchvision.models as models
vgg16=model.vgg16()
transforms模块
用于训练数据的图像预处理