torchvision.transforms，datasets使用方法

最新推荐文章于 2024-07-19 21:10:59 发布

masterShaw

最新推荐文章于 2024-07-19 21:10:59 发布

阅读量368

点赞数

分类专栏： AI 机器学习文章标签： pytorch

本文链接：https://blog.csdn.net/mastershaw/article/details/109998100

版权

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

22 篇文章 0 订阅

订阅专栏

11 篇文章 0 订阅

订阅专栏

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)