Add NAFNet MLX weights (faithful port of megvii-research/NAFNet)

09d01a2 verified about 20 hours ago

997 Bytes

license: mit
library_name: mlx
pipeline_tag: image-to-image
tags:
  - mlx
  - image-restoration
  - deblurring
  - denoising
  - nafnet
base_model: megvii-research/NAFNet

NAFNet width64 (MLX) — Image deblurring

Apple MLX port of NAFNet (Simple Baselines for Image Restoration, ECCV 2022). Runs on Apple Silicon via MLX.

This checkpoint: GoPro (Image deblurring). width64.

Usage

from nafnet_mlx import NAFNetConfig
from nafnet_mlx.pipeline import load_model, restore_to_file
m = load_model("model.safetensors", NAFNetConfig.gopro_width64())
restore_to_file(m, "input.png", "output.png")

Validation

Faithful NHWC port (SimpleGate, Simplified Channel Attention, channel-axis LayerNorm2d, UNet + PixelShuffle). PT-vs-MLX full-model parity on a real image ~1e-6. Uses NAFNetLocal (TLC) local pooling.

License & attribution

MIT. Derived from megvii-research/NAFNet (MIT). See NOTICE.