millieXie / FusionMamba

FusionMamba: Dynamic Feature Enhancement for Multimodal Image Fusion with Mamba
97 stars 5 forks source link

About training other datasets #12

Open JackieWuuuuuu opened 3 days ago

JackieWuuuuuu commented 3 days ago

I have no problem using the KAIST dataset, but when I use the PET_MRI dataset, the final test result is a black and white image, not a color image. What needs to be modified? I think it is a channel problem. If you can help me, I will be very grateful! My results image Your results image

millieXie commented 2 days ago

Medical images are typically converted from RGB to YCbCr before processing, with the Y channel fused with MRI images. After network training, the fused results are converted back from YCbCr to RGB to produce color images. This approach is commonly used in medical image fusion methods.