制作个人图像分类数据集过程中,查阅资料发现几个问题,在解决的过程中总结!
个人认为这是很通俗易懂详细的过程了
制作个人分类用数据集具体步骤如下:
1、将个人收集的图片归到一个文件夹内如下图:
我个人做实验收集的文件夹data内图如下图
2、建立与图片对应的label文件(标签文件),我label中数据如下:
3、在文件夹内建立py代码文件,本实验代码如下:
import dataset
import numpy as np
import torch
import torch.nn as nn
import torch.optim as optim
from torch.autograd import Variable
from torch.utils.data.dataset import Dataset
import torch.utils.data.dataloader as DataLoader
import torchvision.transforms as transforms
class CustomDataset(Dataset):
def __init__(