Method __init__

pycontrast/datasets/util.py:52–53

(self, angles=[0, 90, 180, 270])

class Rotate(object):
    """rotation"""
    def __init__(self, angles=[0, 90, 180, 270]):
        self.angles = angles

    def __call__(self, img):
        angle = np.random.choice(self.angles)