Mona_Jalal:

        norm_landmarks = transforms.Normalize(0.4949, 0.2165)
        landmarks [landmarks != landmarks] = 0
        landmarks = landmarks.unsqueeze_(0)
        landmarks = norm_landmarks(landmarks)



Also, could you please answer this?
If my landmarks are 4 x and y for each image (4 2D keypoints), for Normalize should I have 2 values for mean or one value? If I need to have two values for mean, how is it even actually working with having only one value for mean? Sorry for the existential question!