Skip to content

Wrong input of CANNET #13

@CnChengGong

Description

@CnChengGong

Bread/train_CAN.py

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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions