From 10b9b6767ffc3dce0dac417d55079d158dd4b227 Mon Sep 17 00:00:00 2001 From: Nathan Qin Date: Wed, 27 May 2020 21:32:59 -0600 Subject: [PATCH] new --- data_loader.py | 4 ++++ 1 file changed, 4 insertions(+) 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