Before adding refinement layer

natanielruiz

2017-09-08 abe876183052e9da9c3d633e41386c5c1f4fc1e6

 code/datasets.py

@@ -63,11 +63,11 @@
        rnd = np.random.random_sample()
        if rnd < 0.5:
            if roll >= 0:
                img = img.rotate(60)
                roll -= 60
                img = img.rotate(30)
                roll -= 30
            else:
                img = img.rotate(-60)
                roll += 60
                img = img.rotate(-30)
                roll += 30

        # Bin values
        bins = np.array(range(-99, 102, 3))