def __getitem__()

in mapillary_sls/datasets/msls.py [0:0]

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

    def __getitem__(self, idx):

        # get triplet
        triplet, target = self.triplets[idx]

        # get query, positive and negative idx
        qidx = triplet[0]
        pidx = triplet[1]
        nidx = triplet[2:]

        # load images into triplet list
        output = [torch.stack([self.transform(Image.open(im)) for im in self.qImages[qidx].split(',')])]
        output.append(torch.stack([self.transform(Image.open(im)) for im in self.dbImages[pidx].split(',')]))
        output.extend([torch.stack([self.transform(Image.open(im)) for im in self.dbImages[idx].split(',')]) for idx in nidx])

        return torch.cat(output), torch.tensor(target)