HairMapper icon indicating copy to clipboard operation
HairMapper copied to clipboard

About loss hair and loss face different from paper

Open KIM7AZEN opened this issue 9 months ago • 0 comments

in parer loss hair is difine as

image

but in train_mapper.py this loss do not dot with hair mask.. wondering why

    loss_l2_latent = self.latent_l2_loss(w_hat, res_w)
    loss_dict['loss_l2_latent'] = float(loss_l2_latent)
    loss += loss_l2_latent * self.latent_l2_lambda

    loss_l2_img = torch.mean(((res_x - x_hat)) ** 2, dim=[0, 1, 2, 3])  ### do not dot with hair mask
    loss_dict['loss_l2_res_img'] = float(loss_l2_img)
    loss += loss_l2_img * self.img_l2_lambda_res

    loss_l2_img = torch.mean(((origin_img - x_hat) * mask) ** 2, dim=[0, 1, 2, 3])
    loss_dict['loss_l2_origin_img'] = float(loss_l2_img)
    loss += loss_l2_img * self.img_l2_lambda_origin

KIM7AZEN avatar May 07 '24 04:05 KIM7AZEN