百度网盘AI大赛:文档图像摩尔纹消除第二名方案
在百度网盘AI大赛文档图像摩尔纹消除赛题中,团队采用改进的多尺度卷积神经网络(基于IDR网络改动),在A榜获第一、B榜获第二。主要改动包括叠加Non-Local模块、增加特征通道,配合数据增强和调参优化,测试集B后处理可提升PSNR 1db。

去除摩尔纹,A榜第一,B榜第二。
采用改进的多尺度卷积神经网络来去除摩尔纹。 多尺度网络基于IDR: Self-Supervised Image Denoising via Iterative Data Refinement进行改动。
免费影视、动漫、音乐、游戏、小说资源长期稳定更新! 👉 点此立即查看 👈
一、任务分析
本次比赛去摩尔纹,通过消除拍摄时的摩尔纹噪声,还原图片原本的样子,其本质,也是图像复原任务。从评价指标来看,主要是PSNR和MSSSIM,也是图像复原任务中常用的客观评价指标,并且指标归一化后也是很合理的,PSNR和MSSSIM占比一样。其次,从本次的比赛训练数据和测试数据分析。训练数据有1000对,其中包含典型摩尔纹的场景数据占比较少,更多的数据场景,可以归为色彩增强。但是测试集A和测试集B中,典型摩尔纹场景的图片占比较大,因此在这批训练数据上训练的模型,在测试集中会有不稳定的差异。解决方案来讲,最直观有效的是增加对应的训练数据。其次就是进行后处理。

二、模型构建思路及调优过程
(1)算法思路;
基于对本次训练数据的分析,我们团队没有直接尝试经典的去摩尔纹的网络。而是修改了几个去噪网络来进行任务。最初是对去年性能很好的MIMOUNet,Restormer以及Uformer网络进行修改。其中MIMOUNet只进行三次降采样,训练和修改的效果来看,线上没有训练很高。。而基于Transformer的方案网络存在训练和测试,效果差异大,不稳定的情况,因此舍弃了使用Transformer的方案(这里不排除是我训练的不够好的情况,各位如果有训练效果还可以的,欢迎交流)。后来对比赛任务进一步分析,感觉无论是色彩在增强还是去摩尔纹,不同的图片的退化差异都比去噪任务更大,需要比去噪更大的感受野。因此就想尝试更多次的降采样,于是使用paddle实现了IDR网络。初次的训练线上分数0.66,提分很明显。之后在该网络的底层增加Non-Local模型,进一步增大感受野。指标也有了进一步的提升。
主要改动:
基于IDR网络:
1.在底层叠加了Non-Local模块,提高网络获取全局信息的能力。
2.把网络特征通道从48增加到96,提高网络的学习能力。
(2)数据增强/清洗策略;
1.训练数据增强:水平翻转,竖直翻转,旋转。
2.测试增强:水平翻转,竖直翻转。
(3)调参优化策略;
每30w iter,lr减半
(4)训练脚本/代码
python train.py
(5)测试脚本/代码,必须包含评估得到最终精度的运行日志; python test.py
三、后处理流程:

**说明:**测试集B更适合后处理,PSNR提升应该在1db,但是由于需要手动设置阈值,未使用该方式。
四、代码内容说明
checkpoint: 保存模型的文件夹
dataloading: 定义数据加载
modules: 定义模型
log: 训练日志
loss: 损失函数
相关攻略
常见报错解析:“Access Not Configured”故障排除指南 许多开发者和团队成员在使用OpenClaw集成飞书时,都曾遭遇过一个典型的中断提示:“access not configured”(访问未配置)。该提示会明确显示您的飞书账户ID及一组唯一的配对验证码,并指出需要联系机器人所有
OpenClaw 常用指令大全与使用详解 openclaw status:此命令是查看OpenClaw系统整体健康状态的核心指令,执行后即获取服务运行状况的全面报告,是日常运维的首要诊断工具。 openclaw gateway restart:在修改网关配置后,必须运行此指令以重启网关服务,使配置文
如何通过 OpenClaw 实现 Chrome 浏览器自动化操控 在软件开发与自动化测试领域,持续学习是常态。本文旨在详细介绍如何利用 OpenClaw 连接并控制一个已开启的 Chrome 浏览器实例,实现点击、文本输入、文件上传、页面滚动、屏幕截图以及执行 JavaScript 等自动化操作。整
项目概述 你是否希望将强大的 AI 助手带入日常聊天?本教程将指导你完成搭建流程,让你能在 QQ 上直接调用 OpenClaw 智能助手,实现无门槛的 AI 对话体验。 架构说明 ┌─────────────┐ ┌──────────────┐ ┌─────────────┐ │ QQ 用户 │ ─
一 下载并安装Node js,全程保持默认设置 首先,请前往Node js官方网站的下载中心:https: nodejs org zh-cn download。根据您的操作系统(Windows Mac Linux)下载对应的安装程序。运行安装向导时,整个过程非常简单,您只需连续点击“下一步”按钮
热门专题
热门推荐
速览攻略:世界圣羽翼王核心打法与全面解析 本攻略将为你完整呈现《洛克王国》世界圣羽翼王的通关秘籍,深度剖析两种高效实战打法:追求极致速度的“燃薪虫四回合速通”与稳定输出的“酷拉无限连击流”。文章将进一步解析这位翼系精灵王的技能机制、属性克制关系及其在PVE与PVP中的实战定位,帮助你彻底掌握应对其隐
速览:工程系统核心机制解析 在《异种航员2》中,工程系统是整个抵抗力量赖以运转的“战略后勤中枢”。无论是研发新武器、生产重型装甲还是制造先进飞行器,所有实体装备的产出都依赖于此。简言之,该系统的核心运作围绕着两大关键:工程师人力的高效配置与全球稀缺资源的精细化调度。工程师的数量直接决定了每个项目的建
核心速览 在《洛克王国世界》中,治愈兔是一位兼具功能性任务角色与实战辅助能力的精灵。它的价值不仅在剧情推进中体现,更在于对战里出色的治疗与防护表现。本文将为你全面解析治愈兔的精准获取位置、种族属性特点以及实战技能搭配,助你顺利捕捉并最大化其在队伍中的作用。所有关键信息将通过清晰的图文内容详细展示,确
速览 在《红色沙漠》中,挑战传说之狼这一强大的任务BOSS,需要玩家进行充分的准备并遵循完整的任务流程。整个过程环环相扣,你必须首先参与塞莱斯特家族的势力任务,通过完成任务将家族声望提升至指定等级,才能解锁【传说之狼】的专属讨伐任务,最终直面这个传说中的强大生物。 红色沙漠传说之狼怎么打 归根结底,
【宝可梦Pokopia】舒适度全解析:快速提升环境等级的核心秘诀 你是否正在探索《宝可梦Pokopia》世界,并希望有效提升宝可梦栖息地的舒适度?舒适度不仅是衡量宝可梦快乐程度的晴雨表,更是解锁游戏核心内容、加速发展的关键驱动指标。本攻略将系统性地为你揭示提升舒适度的核心途径,涵盖从装饰栖息地、建造





