CBDNet-火炬
CBDNet的非官方PyTorch实施。
更新
2021.04.02:该代码可能存在一些实现错误,请谨慎使用。 拉请求是受欢迎的。 2020.12.04:我们使用新的数据集训练了CBDNet模型。 PSNR(DND基准):38.06-> 39.63。
快速开始
从下载数据集和预训练的模型。
将文件解压缩到data文件夹和save_model文件夹,如下所示:
~/
data/
SIDD_train/
... (scene id)
Syn_train/
... (id)
DND/
images_srgb/
... (mat files)
... (mat files)
save_model/
checkpoint.pth.tar
训练模型:
python tr
1