新闻中心

百度网盘AI大赛:文档图像摩尔纹消除第二名方案

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

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

百度网盘ai大赛:文档图像摩尔纹消除第二名方案 -

去除摩尔纹,A榜第一,B榜第二。

采用改进的多尺度卷积神经网络来去除摩尔纹。 多尺度网络基于IDR: Self-Supervised Image Denoising via Iterative Data Refinement进行改动。

一、任务分析

本次比赛去摩尔纹,通过消除拍摄时的摩尔纹噪声,还原图片原本的样子,其本质,也是图像复原任务。从评价指标来看,主要是PSNR和MSSSIM,也是图像复原任务中常用的客观评价指标,并且指标归一化后也是很合理的,PSNR和MSSSIM占比一样。其次,从本次的比赛训练数据和测试数据分析。训练数据有1000对,其中包含典型摩尔纹的场景数据占比较少,更多的数据场景,可以归为色彩增强。但是测试集A和测试集B中,典型摩尔纹场景的图片占比较大,因此在这批训练数据上训练的模型,在测试集中会有不稳定的差异。解决方案来讲,最直观有效的是增加对应的训练数据。其次就是进行后处理。

百度网盘AI大赛:文档图像摩尔纹消除第二名方案 -

二、模型构建思路及调优过程

(1)算法思路;
基于对本次训练数据的分析,我们团队没有直接尝试经典的去摩尔纹的网络。而是修改了几个去噪网络来进行任务。最初是对去年性能很好的MIMOUNet,Restormer以及Uformer网络进行修改。其中MIMOUNet只进行三次降采样,训练和修改的效果来看,线上没有训练很高。。而基于Transformer的方案网络存在训练和测试,效果差异大,不稳定的情况,因此舍弃了使用Transformer的方案(这里不排除是我训练的不够好的情况,各位如果有训练效果还可以的,欢迎交流)。后来对比赛任务进一步分析,感觉无论是色彩在增强还是去摩尔纹,不同的图片的退化差异都比去噪任务更大,需要比去噪更大的感受野。因此就想尝试更多次的降采样,于是使用paddle实现了IDR网络。初次的训练线上分数0.66,提分很明显。之后在该网络的底层增加Non-Local模型,进一步增大感受野。指标也有了进一步的提升。
主要改动:
基于IDR网络:
1.在底层叠加了Non-Local模块,提高网络获取全局信息的能力。
2.把网络特征通道从48增加到96,提高网络的学习能力。百度网盘AI大赛:文档图像摩尔纹消除第二名方案 -

(2)数据增强/清洗策略;
1.训练数据增强:水平翻转,竖直翻转,旋转。
2.测试增强:水平翻转,竖直翻转。

(3)调参优化策略;
每30w iter,lr减半

美图云修 美图云修

商业级AI影像处理工具

美图云修 50 查看详情 美图云修

(4)训练脚本/代码
python train.py

(5)测试脚本/代码,必须包含评估得到最终精度的运行日志; python test.py

三、后处理流程:

百度网盘AI大赛:文档图像摩尔纹消除第二名方案 -

**说明:**测试集B更适合后处理,PSNR提升应该在1db,但是由于需要手动设置阈值,未使用该方式。

四、代码内容说明

checkpoint: 保存模型的文件夹
dataloading: 定义数据加载
modules: 定义模型
log: 训练日志
loss: 损失函数

以上就是百度网盘AI大赛:文档图像摩尔纹消除第二名方案的详细内容,更多请关注其它相关文章!


# 不稳定  # 北仑网站推广企业电话  # 网销怎么自建网站推广页  # 淮阴网站优化软件  # 营销推广的英语  # 台州手机网站建设案例  # 旅行网站建设规定  # 市北区网站建设有哪些  # 贾汪区推广网络营销中心  # 常州网站建设答疑解惑  # seo的定义与关系  # 官网  # python  # 线上  # 更大  # 后处理  # 美图  # 一言  # 文档  # 中文网  # 百度网  # 百度  # 百度网盘  # ai 


相关栏目: 【 行业资讯67740 】 【 技术百科0 】 【 网络运营39195


相关推荐: 如何管理员打开cmd命令行窗口  python和typescript学哪个  苹果16系统多了哪些  如何将系统移到固态硬盘  mac 如何启动命令行模式  为什么用typescript  学typescript有什么用  单片机怎么判定高电平  如何在一串数字前面去掉四位数的命令  市盈率为负数是什么意思  春运什么时候开始抢票  统计学中power值是什么意思  春运抢票哪个平台好一点  苹果16系统有哪些功能  如何创建sql命令  固态硬盘如何保存  市盈率中1stdv是什么意思  typescript和node学哪个  光刻机的作用及工作原理  mac如何使用vi命令行  爱奇艺vip会员可以同时几个人用?  drawing是什么意思  linux如何切换到命令行模式  春运抢票最快几天能成功  安全的ao3镜像网站链接入口  单片机怎么连接电路图  j*a数组求和怎么算  什么软件能下载夸克视频  命令行如何启动应用程序  折叠屏手机哪个牌子性价比高  市盈率292是什么意思  arp命令如何使用  阿里云手机云盘怎么用_阿里云盘苹果手机怎么用教程  win7如何打开命令行窗口  typescript怎么设置滚动条  如何自己加装固态硬盘  typescript怎么添加css样式  typescript和es6先学哪个  新固态硬盘如何装系统  typescript学多久可以学会  如何通过dos命令  如何体验苹果16系统  如何用命令连接mysql  typescript解决了什么  苹果16有哪些改善  如何打开命令框  如何打开命令提示符  苹果16日发售哪些机型  vi命令如何退出  喇叭上标的power30w是什么意思 

搜索