Added new correct cropping for training and smoothing for video.

natanielruiz

2017-08-08 18a21d4b07c581a8954b08518115fb035c712b28

 code/utils.py

@@ -70,5 +70,11 @@
    pose_params = pre_pose_params[:3]
    return pose_params

def get_pt2d_from_mat(mat_path):
    # Get 2D landmarks
    mat = sio.loadmat(mat_path)
    pt2d = mat['pt2d']
    return pt2d

def mse_loss(input, target):
    return torch.sum(torch.abs(input.data - target.data) ** 2)