Ejemplo n.º 1
0
def save_landmark(self, arr, filename,landmark_num, width, height):
        
    temp = np.zeros((height, width))
    x = arr[np.arange(0, landmark_num*2-1, 2)]
    y = arr[np.arange(1, landmark_num*2, 2)]
    temp [x, y] = 255
    from src.utils.util import save_slice
    save_slice(temp, filename)
Ejemplo n.º 2
0
def save_landmarks(self, arr, filename, landmark_num, width, height):
    from src.utils.util import save_slice
    temp = np.zeros((height, width))
    x = arr[:, np.arange(0, landmark_num*2-1, 2)].reshape(-1, 1)
    y = arr[:, np.arange(1, landmark_num*2, 2)].reshape(-1, 1)
    temp[x, y] = 255
    save_slice(temp, filename)