b77efb1269d87560bc9f9e6016bcf77c53557448,examples/data_augmentation.py,DummyDataset,__getitem__,#DummyDataset#Any#,33

Before 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
        target[:2].mul_(512 / ow)
        target[-2:].mul_(512 / oh)

After Change

    def __getitem__(self, idx):
        // get data sample
        sample = self.data_index[idx]

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

Instances