datagen.py 文件源码

def _augment(self,img, hm, max_rotation = 30):
        """ # TODO : IMPLEMENT DATA AUGMENTATION 
        """
        if random.choice([0,1]): 
            r_angle = np.random.randint(-1*max_rotation, max_rotation)
            img =   transform.rotate(img, r_angle, preserve_range = True)
            hm = transform.rotate(hm, r_angle)
        return img, hm

    # ----------------------- Batch Generator ----------------------------------