search

Linfeng-Tang / SwinFusion

This is official Pytorch implementation of "SwinFusion: Cross-domain Long-range Learning for General Image Fusion via Swin Transformer"
161 stars 20 forks source link

test的输出结果问题 #29

Open ymyj opened 1 month ago

ymyj commented 1 month ago

MIF在test运行后的输出PET_MRI结果图仍然为灰度图，我想知道做哪些修改可以让输出和作者您论文中的彩色结果保持一致，期待你的回复，十分感谢

ymyj commented 1 month ago

MIF在测试运行后的输出PET_MRI结果图仍然为灰度图，我想知道做哪些修改可以让输出和作者您论文中的彩色结果保持一致，期待你的回复，十分感谢

41 41 41

Linfeng-Tang commented 4 weeks ago

实验中是使用了YCbCr来进行后处理的 你也可以在融合图像保存代码之后加一个im2RGB 的操作 函数如下 其中f_name表示融合图像的保存路径 vi_name 可以使用PET图像的路径即可 def img2RGB(f_name, vi_name): scale = 32 vi_img = Image.open(vi_name) vi_img = vi_img.convert('YCbCr')

f_img = Image.open(f_name).convert('L')
w, h = f_img.size

vi_img = vi_img.resize((w, h), PIL.Image.ANTIALIAS)
vi_Y, vi_Cb, vi_Cr = vi_img.split()
f_img = Image.open(f_name).convert('L')
f_img = Image.merge('YCbCr', (f_img, vi_Cb, vi_Cr))
f_RGB = f_img.convert('RGB')
f_RGB.save(f_name)