tools.py 文件源码

def rotatehm(hm, angle):
    """
        Given a heatMap, returns a rotated heatMap
        args : 
            hm      : (numpy.array) heatMap
            angle : (int) Angle
    """
    rot_hm = np.zeros((16,64,64))
    for i in range(16):
        rot_hm[i] = transform.rotate(hm[i],angle)
    return rot_hm