赞
踩
FusionGAN是使用生成对抗式神经网络(GAN)来完成红外和可见光图像融合任务,并且取得了很好的实验效果。
FusionGAN论文地址:
FusionGAN: A generative adversarial network for infrared and visible image fusion
FusionGAN代码已经开源在GitHub上FusionGAN代码
代码下载好,解压后,应该有如下内容
cheatpoint文件夹中CGAN_120文件夹内存放的是预训练模型
Train_ir/Train_vi分别存放的是训练用的红外和可见光数据集
Test_ir/Test_vi分别存放的是测试用的红外和可见光数据集
如果只需要使用预训练模型运行代码,只需要运行test_one_image.py文件即可,运行代码为
python3 test_one_image.py
这里使用python3是因为我使用的环境原因,如果您运行报错可尝试使用如下代码
python test_one_image.py
运行正确后,
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。