Skip to content

torchvision 参考

译者:BXuan694

torchvision 包收录了若干重要的公开数据集、网络模型和计算机视觉中的常用图像变换

包参考

torchvision.get_image_backend()

查看载入图片的包的名称

torchvision.set_image_backend(backend)

指定用于载入图片的包

参数: backend (string) – 图片处理后端的名称,须为{'PIL', 'accimage'}中的一个。accimage包使用了英特尔IPP库。这个库通常比PIL快,但是支持的操作比PIL要少。


回到顶部