AI ISP训练数据生成调研

1. 概念

sRGB图和raw图是两种不同的图像格式,它们之间的转换需要一些特定的算法。
sRGB图是经过图像信号处理 (ISP) 的彩色图像,它包含了白平衡、去马赛克、降噪、颜色空间转换、色调映射等操作。
raw图是未经处理的原始图像,它保存了相机传感器捕获的原始数据,通常有更高的位深度和动态范围。

2. 方法分类

  • 基于模型的方法:这类方法利用了相机ISP中的各个模块对其进行建模,生成合成的训练数据对。例如,Unprocessing Images for Learned Raw Denoising 提出了一种方法’unprocess’, 考虑到相机ISP中的各个模块对其进行建模,生成syntactic训练数据对。这种方法可以批量生成raw视频数据,并且根据自己的噪音模型往上加噪音。但是,这种方法也存在一些问题,比如rgb的8bit和raw的12bit的问题,以及tone mapping和白平衡等非线性操作会对数据集造成退化。

  • 基于可逆网络的方法:这类方法利用了可逆网络结构来学习RAW和sRGB之间的双向映射,从而实现可逆的ISP。例如,Invertible Image Signal Processing 重新设计了一个端到端可逆ISP,可以绕过传统的模块,最大限度地减少 raw 数据和 jpeg 图像转换过程中的信息损失,进一步实现高质量 raw 数据重建。这种方法可以学习到ISP pipeline的一些非线性特性,并且对JPEG压缩鲁棒。但是,这种方法也需要大量的训练数据,并且可能存在一些不可逆的情况。

  • 基于循环一致性的方法:这类方法利用了循环一致性损失来约束RAW和sRGB之间的转换,从而实现无监督或半监督的ISP。例如,CycleISP: Real Image Restoration via Improved Data Synthesis 提出了一种循环ISP框架,它可以从真实sRGB图像合成高质量的RAW图像,并且可以从合成或真实RAW图像恢复高质量的sRGB图像。这种方法可以利用大量的真实sRGB图像作为数据源,并且可以改善合成数据集中存在的信息缺失问题。但是,这种方法也需要设计合适的循环一致性损失,并且可能存在一些循环伪影。

3. 数据集

4. 评价指标

  • 峰值信噪比 (PSNR):这是一种常用的图像质量评价指标,它通过计算原始图像和重建图像之间的均方误差 (MSE) 来衡量两者的相似度。PSNR越高,表示重建图像越接近原始图像。例如,Unprocessing Images for Learned Raw Denoising 使用了PSNR来评价从合成的sRGB图重建raw图的性能。
  • 结构相似性指数 (SSIM):这是一种考虑了人类视觉感知特性的图像质量评价指标,它通过计算原始图像和重建图像之间的亮度、对比度和结构相似度来衡量两者的视觉质量。SSIM越接近1,表示重建图像越接近原始图像。例如,Invertible Image Signal Processing 使用了SSIM来评价从JPEG压缩的sRGB图重建raw图的性能。
  • 均方根色彩误差 (RMSE-C):这是一种专门用于评价颜色还原能力的指标,它通过计算原始图像和重建图像之间的色彩距离来衡量两者的颜色相似度。RMSE-C越低,表示重建图像越接近原始图像的颜色。例如,CycleISP: Real Image Restoration via Improved Data Synthesis 使用了RMSE-C来评价从真实sRGB图合成raw图的性能。

根据我搜索的结果,这些评价指标各有优缺点,简要如下:

图像质量评价指标的选择取决于不同的应用场景和目标。一般来说,有以下几个原则:

  • 如果有原始图像作为参考,可以选择全参考方法,如PSNR, SSIM等,这些方法可以比较直观地反映两幅图像之间的差异程度。
  • 如果没有原始图像作为参考,可以选择无参考方法,如BRISQUE, NIQE等,这些方法可以根据图像的自然度或统计特征来评估图像的质量。
  • 如果要考虑人类视觉系统的特性,可以选择基于感知的方法,如VIF, FSIM等,这些方法可以模拟人眼对图像的感知过程,更符合主观评价。
  • 如果要评价特定类型的失真,如模糊,噪声,压缩等,可以选择针对性的方法,如BLIINDS, DIIVINE等,这些方法可以针对不同的失真因素进行优化。
  • 如果要评价多种类型的失真,或者不清楚失真类型,可以选择通用的方法,如MAD, GM-LOG等,这些方法可以适应多种场景和失真情况。
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,163评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,301评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,089评论 0 352
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,093评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,110评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,079评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,005评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,840评论 0 273
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,278评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,497评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,667评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,394评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,980评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,628评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,796评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,649评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,548评论 2 352

推荐阅读更多精彩内容