diff --git a/data_loader.py b/data_loader.py index 4062072..3d4bfa1 100644 --- a/data_loader.py +++ b/data_loader.py @@ -85,6 +85,10 @@ class RandomCrop(object): def __call__(self,sample): imidx, image, label = sample['imidx'], sample['image'], sample['label'] + if random.random() >= 0.5: + image = image[::-1] + label = label[::-1] + h, w = image.shape[:2] new_h, new_w = self.output_size