natanielruiz
2017-09-08 0b8e19c1cc8ad03805d4ca68f32df6e4806a36e8
code/datasets.py
@@ -60,14 +60,14 @@
            img = img.transpose(Image.FLIP_LEFT_RIGHT)
        # Rotate?
        rnd = np.random.random_sample()
        if rnd < 0.5:
            if roll >= 0:
                img = img.rotate(30)
                roll -= 30
            else:
                img = img.rotate(-30)
                roll += 30
        # rnd = np.random.random_sample()
        # if rnd < 0.5:
        #     if roll >= 0:
        #         img = img.rotate(30)
        #         roll -= 30
        #     else:
        #         img = img.rotate(-30)
        #         roll += 30
        # Bin values
        bins = np.array(range(-99, 102, 3))