FusionGAN图像融合代码学习

FusionGAN简介

FusionGAN是使用生成对抗式神经网络(GAN)来完成红外和可见光图像融合任务,并且取得了很好的实验效果。

论文地址

FusionGAN论文地址:
FusionGAN: A generative adversarial network for infrared and visible image fusion

FusionGAN代码

FusionGAN代码已经开源在GitHub上FusionGAN代码

FusionGAN开源代码内容

代码下载好,解压后,应该有如下内容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

运行正确后,
使用预训练模型测试数据集
实验结果会保存在result文件夹中

使用FusionGAN训练自己的数据集

训练过程

数据集相关

红外和可见光图像训练数据集分别放在Train_ir/Train_vi中,代码默认图像必须是.bmp格式,如果数据集不是.bmp格式,则需要修改utils.py文件第72行代码,修改.bmp为你数据集中的图片格式

  if FLAGS.is_train:
    filenames = os
  • 27
    点赞
  • 138
    收藏
    觉得还不错? 一键收藏
  • 63
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 63
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值