Skip to content

[AI] Add neural restore lighttable module for AI denoise and upscale #20523

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
andriiryzhkov wants to merge 10 commits into
base: master
Choose a base branch
from
+2,887 −0
Open

[AI] Add neural restore lighttable module for AI denoise and upscale #20523

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
0fe3eb9
Add neural restore lighttable module for AI denoise and upscale
andriiryzhkov Mar 14, 2026
012f260
Add output parameters, ICC profile, EXIF, and variable-based output p…
andriiryzhkov Mar 15, 2026
6379e10
Extract AI restore processing to src/ai/restore with reusable API
andriiryzhkov Mar 15, 2026
f83e6b6
Move restore to src/common/ai and compile in main library to fix Wind…
andriiryzhkov Mar 16, 2026
57bee18
Preserve wide-gamut colors in AI denoise/upscale pipeline
andriiryzhkov Mar 20, 2026
fa06d3e
Fix enum redeclaration error by removing comments
andriiryzhkov Mar 21, 2026
83c2162
Allow concurrent AI jobs, block only per-image duplicates
andriiryzhkov Mar 21, 2026
2454c6f
Reuse AI model across jobs with ref-counted restore context
andriiryzhkov Mar 21, 2026
1dffbaa
Move neural restore module down
andriiryzhkov Mar 21, 2026
b0441b5
Simplify wide-gamut preservation to direct gamma pass-through
andriiryzhkov Mar 21, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions data/ai_models.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,22 @@
"task": "mask",
"github_asset": "mask-object-segnext-b2hq.dtmodel",
"default": false
},
{
"id": "denoise-nind",
"name": "denoise nind",
"description": "UNet denoiser trained on NIND dataset",
"task": "denoise",
"github_asset": "denoise-nind.dtmodel",
"default": true
},
{
"id": "upscale-bsrgan",
"name": "upscale bsrgan",
"description": "BSRGAN 2x and 4x blind super-resolution",
"task": "upscale",
"github_asset": "upscale-bsrgan.dtmodel",
"default": true
}
]
}
1 change: 1 addition & 0 deletions src/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -462,6 +462,7 @@ if(USE_AI)
FILE(GLOB SOURCE_FILES_AI
"common/ai_models.c"
"common/ai/segmentation.c"
"common/ai/restore.c"
"develop/masks/object.c"
"gui/preferences_ai.c"
)
Expand Down
3 changes: 3 additions & 0 deletions src/ai/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,8 @@ pkg_check_modules(JSON_GLIB REQUIRED json-glib-1.0)
# GTK3/RSVG needed for headers only (common/darktable.h -> utility.h -> gtk/gtk.h, librsvg/rsvg.h)
pkg_check_modules(GTK3 REQUIRED gtk+-3.0)
pkg_check_modules(RSVG2 REQUIRED librsvg-2.0)
# lcms2 needed for headers (common/colorspaces.h included via darktable.h)
pkg_check_modules(LCMS2 REQUIRED lcms2)

# Include current directory for header
target_include_directories(darktable_ai PUBLIC
Expand All @@ -26,6 +28,7 @@ target_include_directories(darktable_ai PUBLIC
${JSON_GLIB_INCLUDE_DIRS}
${GTK3_INCLUDE_DIRS}
${RSVG2_INCLUDE_DIRS}
${LCMS2_INCLUDE_DIRS}
)

target_link_directories(darktable_ai PUBLIC
Expand Down
Loading
Loading