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)
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)