torchvision.transforms，datasets使用方法

PyTorch数据集划分

最新推荐文章于 2025-10-10 18:28:14 发布

原创最新推荐文章于 2025-10-10 18:28:14 发布 · 421 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

机器学习同时被 2 个专栏收录

22 篇文章

订阅专栏

11 篇文章

订阅专栏

本文介绍了一个用于PyTorch的数据集划分类classDatasetSplit，该类继承自PyTorch的Dataset类，能够实现对已有数据集进行指定索引的子集划分。

部署运行你感兴趣的模型镜像

https://zhuanlan.zhihu.com/p/200876072

http://noahsnail.com/2020/06/12/2020-06-12-%E7%8E%A9%E8%BD%ACpytorch%E4%B8%AD%E7%9A%84torchvision.transforms/

class DatasetSplit(Dataset):
    """An abstract Dataset class wrapped around Pytorch Dataset class.
    """

    def __init__(self, dataset, idxs):
        self.dataset = dataset
        self.idxs = [int(i) for i in idxs]

    def __len__(self):
        return len(self.idxs)

    def __getitem__(self, item):
        image, label = self.dataset[self.idxs[item]]
        return torch.tensor(image), torch.tensor(label)

您可能感兴趣的与本文相关的镜像