73317ca61e33b42353fc4b3010f586124954ca18,examples/data_augmentation.py,DummyDataset,__getitem__,#DummyDataset#Any#,30

Before Change

        // load data, NOTE: modify by cv2.imread(...)
        image = torch.rand(3, 240, 320)
        label = torch.rand(1, 240, 320)
        return dict(images=image, labels=label)

////////////////////////////////////////////////////////////////
// 2. Define the data augmentation operations
// Thanks to the `kornia` design all the operators can be placed inside inside a `nn.Sequential`.

After Change

    def __getitem__(self, index):

        img = Image.open(self.images[index]).convert("RGB")
        target = torch.load(self.targets[index])

        ow, oh = img.size

        // Reescale boxes

Instances