Skip to content

Add rawdenoise-nind model: UtNet2 raw denoiser (Bayer + linear Rec.2020)#20

Merged
TurboGit merged 6 commits intodarktable-org:masterfrom
andriiryzhkov:rawnind
Apr 21, 2026
Merged

Add rawdenoise-nind model: UtNet2 raw denoiser (Bayer + linear Rec.2020)#20
TurboGit merged 6 commits intodarktable-org:masterfrom
andriiryzhkov:rawnind

Conversation

@andriiryzhkov
Copy link
Copy Markdown
Contributor

Summary

Adds a new model package – rawdenoise-nind – providing the raw-denoise task. Two UtNet2 variants ship in a single type: multi package, auto-dispatched by darktable based on the sensor CFA pattern:

  • Bayer – joint denoise + demosaic for Bayer sensors. 4ch packed RGGB input → 3ch camRGB at 2× spatial resolution (the PixelShuffle output head upsamples by 2). Output is in the camera's native RGB space at arbitrary learned gain; the consumer scalar-gain-matches and applies the ColorMatrix.
  • Linear – pure denoise for non-Bayer sensors (X-Trans, Foveon) and anything already demosaicked. 3ch linear Rec.2020 → 3ch linear Rec.2020, same spatial, arbitrary gain.

Both are the canonical base variants from the graph_denoise_models_definitions.yaml map in the upstream training code, with funit=32, activation=LeakyReLU, and match_gain: output (training deliberately leaves the output at an arbitrary scale so the consumer gain-matches at inference).

Source & provenance

OSAID / MOF status

Fully open: model weights GPL-3.0, training code GPL-3.0, training data CC BY 4.0 / CC0, training-run logs published. Qualifies as OSAID v1.0 Open Source AI and MOF Class I (Open Science).

Copy link
Copy Markdown
Member

@TurboGit TurboGit left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@TurboGit
Copy link
Copy Markdown
Member

@andriiryzhkov : One of the CI check is failing, need some more love :) TIA.

@andriiryzhkov
Copy link
Copy Markdown
Contributor Author

@TurboGit : Fixed - demo part was too heavy for CI.

Copy link
Copy Markdown
Member

@TurboGit TurboGit left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@TurboGit TurboGit merged commit 2ce14dd into darktable-org:master Apr 21, 2026
10 checks passed
@andriiryzhkov andriiryzhkov deleted the rawnind branch April 22, 2026 06:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants