torchvision

torchvision包 包含了目前流行的数据集，模型结构和常用的图片转换工具。