深度学习|Pytorch
文章平均质量分 68
澄鑫
不断重复,百炼成钢!
展开
-
【深度学习|Pytorch】torchvision.datasets.ImageFolder详解
self,root: str,):图片存储的根目录,即存放不同类别图片文件夹的前一个路径。即对加载的这些图片进行的前处理的方式,这里可以传入一个实例化的torchvision.Compose()对象,里面包含了各种预处理的操作。对图片类别进行预处理,通常来说不会用到这一步,因此可以直接不传入参数,默认图像标签没有变换,如果需要进行标签的处理,同样可以传入一个实例化的torchvision.Compose()对象。表示图像数据加载的方式,通常采用默认的加载方式,原创 2024-04-03 12:28:31 · 1426 阅读 · 0 评论 -
【CUDA,TensorRT】cuda,cudnn,TensorRT官方下载链接
各版本cuda下载链接各版本cudnn下载链接原创 2024-01-25 21:53:09 · 131 阅读 · 0 评论 -
Pytorch常用api详解
成为Pytorch大师第一步~原创 2023-09-08 15:54:38 · 822 阅读 · 0 评论