cnstark / pysiddtools

A Toolkit for SIDD Dataset

PySIDDTools

SIDD数据集工具集

简介

TODO

使用方法

• 实例化SIDD对象

BASE_DIR = "<Path to SIDD Full Dataset>"
sidd = SIDD(BASE_DIR)

• 获取SceneInstance个数

len(sidd)

• 获取SceneInstance

使用index获取

sidd[0]

• 筛选SceneInstance

更多筛选条件见sidd.py代码

s6_visible_L = sidd.filter(visible=True, smartphone="S6", luminance="L")

• 读取raw图像

gt_0 = s6_visible_L[0].gt_raw(0)
gt_0_bggr = s6_visible_L[0].gt_raw(0, pattern="BGGR")

noisy_0 = s6_visible_L[0].noisy_raw(0)
noisy_0_rggb = s6_visible_L[0].noisy_raw(0, pattern="RGGB")
noisy_0_channels = s6_visible_L[0].noisy_raw(0, pattern="RGGB", split_channel=True)

• 读取rbg图像

gt_0 = s6_visible_L[0].gt_srgb(0)
noisy_0 = s6_visible_L[0].noisy_srgb(0)

pysiddtools's Issues

mat文件转为dng文件

你好，我下载了SIDD数据集的mat格式的文件，怎么转为DNG格式的啊？

gt_raw和gt_srgb的shape问题

您好，我在运行的时候发现如果raw图像的Orientation为1的话，gt_raw读出来的大小是W H，而gt_srgb读出来的大小是 3 H W，这是正常的吗？如果要训练的话，简单的调换gt_raw的行列就行了吧？感谢！