Pytorch-制作自己的多标签数据集
# 需求 如图所示,想通过 Pytorch 加载自己的数据集,每一张图片后面有三个标签。该如何实现呢?看下面 # 代码 from PIL import Imageimport torchfrom torchvision import transformsimport numpy as npfrom matplotlib import pyplot as plt# 创建自己的类:MyDataset, 这个类是继承的 torch.utils.data.Datasetclass MyDataset(torch.utils.data.Dataset): def __init__(self,...
more...







