Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"
MIT License
210 stars 37 forks source link

Do you use only one channel in FuSionNet ? #16

Open chengyu89527 opened 1 year ago

chengyu89527 commented 1 year ago

def forward(self, image_vis,image_ir):

split data into RGB and INF

    x_vis_origin = image_vis[:,:1]
    x_inf_origin = 

x_vis_origin only get one channel? why?

Linfeng-Tang commented 1 year ago

Yes, we only fusion the Y-channel in our FusionNet .