torchvision.datasets.ImageFolder是pytorch中通用的数据加载器,其加载的数据形式是
数据形式如图所示,并且其会把文件夹自动的转化为0,1,2…等类别号,方便计算梯度,我们自己不用管。
torchvision.datasets.ImageFolder的用法
最新推荐文章于 2024-05-04 11:15:00 发布
torchvision.datasets.ImageFolder是pytorch中通用的数据加载器,其加载的数据形式是
数据形式如图所示,并且其会把文件夹自动的转化为0,1,2…等类别号,方便计算梯度,我们自己不用管。