DataLoader getitem

最新推荐文章于 2024-09-27 10:29:25 发布

It-is-me!

最新推荐文章于 2024-09-27 10:29:25 发布

阅读量985

点赞数

分类专栏： Deep Learning

本文链接：https://blog.csdn.net/weixin_47556699/article/details/106998665

版权

Deep Learning 专栏收录该内容

57 篇文章 1 订阅

订阅专栏

def __getitem__(self, index):

        label = self.labels[index]
        frames_array = np.zeros((frames_length, 3, 200, 200), dtype=np.float32)
        ......

        for idx in range(frames_length):
            frame = Image.open(path[idx]).convert("RGB")
            if self.transform:
                frame = self.transform(frame)
                frame = frame.numpy()
            frames_array[idx, :, :, :] = frame

        ......
        frames_array = frames_array.transpose((1, 0, 2, 3))
        label = torch.tensor(label)
        frames = torch.tensor(frames_array)
        return frames, label