From bf1138509f0010946d7ceb4e92c265c6b223fcbb Mon Sep 17 00:00:00 2001 From: MishaLaskin Date: Wed, 19 Feb 2020 21:48:22 -0800 Subject: [PATCH] rm iaa --- utils.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/utils.py b/utils.py index 5f5a2f4..76514e1 100644 --- a/utils.py +++ b/utils.py @@ -8,7 +8,6 @@ import random from torch.utils.data import Dataset, DataLoader from torchvision import transforms import time -from imgaug import augmenters as iaa from skimage.util.shape import view_as_windows class eval_mode(object): @@ -90,18 +89,6 @@ class ReplayBuffer(Dataset): self.last_save = 0 self.full = False - self.seq = iaa.Sequential([ - # crop images from each side by 0 to 16px (randomly chosen) - iaa.Crop(px=(0, 20)), - #iaa.Fliplr(0.5), - # iaa.Affine( - #scale={"x": (0.8, 1.2), "y": (0.8, 1.2)}, - #translate_percent={"x": (-0.2, 0.2), "y": (-0.2, 0.2)}, - #rotate=(-20, 20), - #shear=(-8, 8) - #) - - ], random_order=True)