二维码提高对比度文献调研(2)--HINet: Half Instance Normalization Network for Image Restoration

简介

(1)论文名称:
HINet: Half Instance Normalization Network for Image Restoration CVPR 2021
(2)论文链接:
https://arxiv.org/abs/2105.06086
(3)源代码:
https://github.com/megvii-model/HINet

主要idea

这篇论文本文探讨了规范化在低层次视觉任务中的作用,作者提出了一种新的块:半规范化块(HIN块),以提高图像恢复网络的性能。基于HIN块,作者设计了一个简单而强大的多级网络HINet,它由两个子网组成。通过将HIN block用于每个子网络的编码器中,提升特征的稳定性。此外,使用交叉阶段特征融合(CSFF)和监督注意力模块(SAM),用于丰富不同阶段网络之间的特征。
Network Architecture

实验简介

(1)数据集:
作者一共用了四个数据集

  • SIDD:一个图像去噪数据集,包含来自10个不同光照条件下的3万幅噪声图像
  • Gopro:用于去模糊的GoPro数据集由3214张1,280×720大小的模糊图像组成,这些图像分为2103张训练图像和1111张测试图像。
  • REDS:数据集由300个分辨率为720×1,280的视频序列组成,每个视频有100帧,其中训练集、验证集和测试集分别有240、30和30个视频。
  • Synthetic Rain Datasets:由13712张从多个数据集(Rain14000, Rain1800, Rain800, Rain12)收集的干净雨图像对组成。使用单一训练过的模型,可以在各种测试集上执行评估,包括Rain100H、Rain100L、Test100、Test2800和Test1200。

(2)预训练模型:

  • HINet-SIDD-0.5x
  • HINet-SIDD-1x
  • HINet-GoPro
  • HINet-REDS
  • HINet-Rain13k

实验结果

所遇到的问题

  • 代码中引用的scikit-image包必须是0.18版本,0.15版本会缺少相关功能且报错

运行失败信息

GPU不够

  • 本人做实验所使用的电脑显卡为1050ti,但是在运行代码的Single Image Inference(执行操作为:python basicsr/demo.py -opt options/demo/demo.yml)时会报错,显示我的内存不足。具体信息如下图所示:
    RuntimeError
    这里输入为:python basicsr/demo.py -opt options/demo/demo.yml
    使用的模型为:HINet-GoPro.pth
### HiNet:基于可逆网络的深度图像隐藏研究与实现 #### 背景介绍 HiNet 是一种用于图像隐藏的新方法,旨在解决传统图像隐写术中的容量有限、安全性不足以及可见度高的问题。该模型利用了可逆网络的独特性质来构建一个高效的图像隐藏框架[^1]。 #### 技术原理 在 HiNet 中,图像隐藏被设计成一个双向的过程——即前向传播负责将秘密消息嵌入到载体图片中形成含密图;而反向传播则用来从含密图里提取原始的秘密信息并重建载体图片。由于这两个操作共用了同一套权重参数,在完成一次完整的训练之后便能够同时得到两个功能完备的部分[^2]。 为了增强系统的鲁棒性和隐蔽效果,研究人员引入了一个特别定制的小波变换域上的损失函数—低频小波损失算法。这一创新使得即使面对各种攻击手段(如裁剪、压缩),所携带的信息依然可以保持较高的准确性[^3]。 #### 实验验证 实验结果显示,相比于其他同类方案,采用 HiNet 进行数据传输不仅拥有更大的有效载荷能力而且还能更好地保护隐私不被轻易察觉。此外,开源项目提供了详细的代码示例供开发者学习参考[^4]。 ```python import torch.nn as nn class HiNet(nn.Module): def __init__(self, num_channels=3): super(HiNet, self).__init__() # 定义编码器部分 self.encoder = Encoder(num_channels=num_channels) # 定义解码器部分 self.decoder = Decoder(num_channels=num_channels) def forward(self, cover_image, secret_message=None): if self.training: stego_image = self.encoder(cover_image, secret_message) recovered_secret = self.decoder(stego_image) return stego_image, recovered_secret else: recovered_secret = self.decoder(cover_image) return recovered_secret ```
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值