pytorch-torchvision

torchvision是独立于PyTorch的关于图像操作的一个工具库,目前包括六个模块:

  • torchvision.datasets:几个常用视觉数据集,可以下载和加载,以及如何编写自己的Dataset。
  • torchvision.models:经典模型,例如AlexNet、VGG、ResNet等,以及训练好的参数。
  • torchvision.transforms:常用的图像操作,例随机切割、旋转、数据类型转换、tensor与numpy 和PIL Image的互换等。
  • torchvision.ops:提供CV中常用的一些操作,比如NMS、ROI_Align、ROI_Pool等。
  • torchvision.io:提供输入输出的一些操作,目前针对的是视频的写入写出。
  • torchvision.utils:其他工具,比如产生一个图像网格等。

transforms:https://pytorch.org/docs/stable/torchvision/transforms.html#

你可能感兴趣的:(pytorch)