File size: 956 Bytes
3ca8b87 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | ---
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 denoising
Apple MLX port of **[NAFNet](https://github.com/megvii-research/NAFNet)** (Simple Baselines for
Image Restoration, ECCV 2022). Runs on Apple Silicon via [MLX](https://github.com/ml-explore/mlx).
This checkpoint: **SIDD** (Image denoising). width64.
## Usage
```python
from nafnet_mlx import NAFNetConfig
from nafnet_mlx.pipeline import load_model, restore_to_file
m = load_model("model.safetensors", NAFNetConfig.sidd_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.
## License & attribution
MIT. Derived from megvii-research/NAFNet (MIT). See `NOTICE`.
|