def __getitem__()

in pytouch/datasets/sequence.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

    def __getitem__(self, seq_idx):
        sequence_img_list = []
        sequence_paths = [img[0] for img in self.imgs if img[1] == seq_idx]
        for img in sequence_paths:
            sample = self.loader(img)
            if self.transform is not None:
                sample = self.transform(sample)
            sequence_img_list.append(sample)
        # img sequence list -> list of nd tesnsors S x C x H x W
        return torch.stack(sequence_img_list)