pythorch CNN 小教程,关于如何自定义数据集
之前用官方的minist数据集测试过了,现在想加载自己的数据集并进行识别分类,应该如何加载呢?
利用ImageFolder函数,将所需分类的图片提前按文件夹分类分别置于train\ val\文件夹下即可
代码如下所示:
import torch
import torch.nn as nn
import torch.nn.functional as F
import numpy
import...
原创
2019-05-10 08:59:12 ·
1266 阅读 ·
8 评论