Ship the denoise networks with any height and width

mosaic-{fast,medium,best}.onnx are the shipped networks re-exported by
darkroom-denoise tools/export_whole.py (22ea648) from the checkpoints
the 1408 files came from: identical to them at 1408 (max |d| = 0), to
torch at 592 x 848, and to tiled inference over the reflected frame in
f64. 42 MB together. The Arch package and the Windows installer carry
them beside the fixed files; the APK leaves them out, since the Hexagon
takes fixed shapes only.
This commit is contained in:
2026-10-06 21:47:20 -04:00
parent 9cba420fd5
commit 3761281dd1
5 changed files with 24 additions and 8 deletions
+4
View File
@@ -323,6 +323,10 @@ for _dir in face scene inpaint denoise; do
for f in "${ASSETS}"/*; do
case "$(basename "${f}")" in
README.md) continue ;;
# The denoisers' any-size exports run whole frames on TensorRT
# and CUDA (denoise.md §14); the Hexagon takes fixed shapes, and
# 42 MB of graphs it never loads stay out of the APK.
mosaic-fast.onnx | mosaic-medium.onnx | mosaic-best.onnx) continue ;;
esac
cp "${f}" "${OUT}/staging/assets/models/"
_bundled="${_bundled} $(basename "${f}")"