-
Notifications
You must be signed in to change notification settings - Fork 14
Open
Description
Lines 95 to 99 in e0c5068
| texture_en, cb_en, cr_en = torch.split(kornia.color.rgb_to_ycbcr(image / torch.clamp_min(texture_illumi, self.eps)), | |
| 1, dim=1) | |
| texture_gt, cb_gt, cr_gt = torch.split(kornia.color.rgb_to_ycbcr(image_gt), 1, dim=1) | |
| colors = self.model_canet(torch.cat([texture_in, cb_in, cr_in, texture_gt, cb_en, cr_en], dim=1)) |
Should be
colors = self.model_canet(torch.cat([texture_in, cb_in, cr_in, texture_en, cb_en, cr_en], Metadata
Metadata
Assignees
Labels
No labels