火炬深度学习的图像和视频数据集及模型
torch-vision的Python项目详细描述
火炬视野
此存储库包括:
- vision.datasets :流行vision的数据加载器 数据集
- vision.models :流行机型的定义 体系结构,如alexnet、vgg、resnet和pre-trained 型号:
- vision.transforms :常见的图像转换 例如随机裁剪、旋转等。
- vision.utils :一些有用的东西,例如节省张量(3 x h x w)作为图像到磁盘,给定一个小批量创建一个图像网格, < 等