NAFNet: Image restoration without nonlinear activations
Efficient PyTorch architecture for image restoration tasks.
Learn more about NAFNet
NAFNet is a convolutional neural network architecture designed for low-level vision tasks including image deblurring, denoising, and stereo super-resolution. The model is implemented in PyTorch and operates by replacing traditional nonlinear activation functions with multiplication operations or removing them entirely from the network. The architecture achieves competitive results on standard benchmarks such as GoPro for deblurring and SIDD for denoising while requiring substantially fewer computational operations than comparable methods. The framework is applicable to various image restoration scenarios where computational efficiency and inference speed are relevant considerations.
Activation-free design
The architecture removes or replaces nonlinear activation functions (ReLU, GELU, Sigmoid, Softmax) with simpler operations like multiplication, simplifying the model structure while maintaining restoration quality across multiple tasks.
Computational efficiency
Achieves comparable or improved performance on standard benchmarks while requiring significantly fewer multiply-accumulate operations than prior methods, as demonstrated on GoPro and SIDD datasets.
Multi-task capability
Provides implementations for multiple image restoration tasks including image deblurring, image denoising, and stereo image super-resolution through a unified architectural approach with task-specific variants.
Related Repositories
Discover similar tools and frameworks used by developers
Crawl4AI
Async browser automation extracting web content for LLMs.
Stanford Alpaca
Research project that fine-tunes LLaMA models to follow instructions using self-generated training data.
Wan2.1
Diffusion transformer models for text and image-to-video generation.
Video2X
ML-powered video upscaling, frame interpolation, and restoration with multiple backend support.
Summarize
CLI and browser extension that generates summaries from URLs, files, videos, podcasts, and other media sources.