一种图像预处理网络的训练方法和图像预处理方法
申请人信息
- 申请人:中国科学技术大学
- 申请人地址:230026 安徽省合肥市包河区金寨路96号
- 发明人: 中国科学技术大学
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种图像预处理网络的训练方法和图像预处理方法 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311776107.0 |
| 申请日 | 2023/12/22 |
| 公告号 | CN117459737A |
| 公开日 | 2024/1/26 |
| IPC主分类号 | H04N19/19 |
| 权利人 | 中国科学技术大学 |
| 发明人 | 刘东; 叶帆; 李礼 |
| 地址 | 安徽省合肥市包河区金寨路96号 |
摘要文本
本发明提供了一种图像预处理网络的训练方法和图像预处理方法。该图像预处理网络的训练方法包括:利用不同的JPEG量化因子,对训练数据图像集中的原始图像执行JPEG压缩操作;利用图像预处理网络原始图像的预处理图像,利用可微分的JPEG模拟器的码率估计网络计算预处理图像的码率;利用可微分JPEG模拟器对原始图像的预处理结果进行重建,计算重建图像与所对应的原始图像之间的失真;利用预定义的率失真损失函数得到率失真损失值,通过率失真损失值对图像预处理网络进行参数调整;对码率估计网络进行参数调整;通过交替迭代的方式对图像预处理网络和码率估计网络进行参数调整,得到训练完成的图像预处理网络和码率估计网络。 微信公众号专利查询网
专利主权项内容
1.一种图像预处理网络的训练方法,其特征在于,包括:利用不同的JPEG量化因子,对训练数据图像集中的每张原始图像执行JPEG压缩操作,其中,所述JPEG压缩操作用于计算所述训练数据图像集中每张图像的相互独立的拉格朗日系数;利用图像预处理网络对所述训练数据图像集中的每张原始图像进行预处理,得到每张所述原始图像的预处理图像,并利用可微分的JPEG模拟器的码率估计网络计算每张所述预处理图像的码率,其中,所述可微分的JPEG模拟器包括码率估计网络、DCT变换模块、逆DCT变换模块和自适应量化DCT域掩膜模块;利用所述可微分JPEG模拟器对每张所述原始图像的预处理结果进行重建,得到每张所述原始图像的重建图像,并计算每张所述重建图像与所对应的原始图像之间的失真;利用预定义的率失真损失函数处理每张所述原始图像的拉格朗日系数、每张所述预处理图像的码率以及所述失真,得到率失真损失值,并通过最小化所述率失真损失值对所述图像预处理网络进行参数调整,直到满足预设训练条件,得到训练完成的图像预处理网络。