def __getitem__()

in src/similarity/siamese.py [0:0]

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

    def __getitem__(self, idx):
        
        img1_name = os.path.join(self.root_dir, self.index.iloc[idx, 0])
        img1 = Image.open(img1_name)
        
        img2_name = os.path.join(self.root_dir, self.index.iloc[idx, 1])
        img2 = Image.open(img2_name)
        
        image1_tensor = self.transform(img1)
        image2_tensor = self.transform(img2)

        label = self.index.iloc[idx, 2]
        
        return {"img1": image1_tensor, "img2": image2_tensor, "label": label}