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.
Top in AI & ML
Related Repositories
Discover similar tools and frameworks used by developers
Wan2.2
Open-source diffusion framework for multi-modal video generation.
X Recommendation Algorithm
Open source implementation of X's recommendation algorithm for timeline and notification ranking.
GroundingDINO
Zero-shot object detection from text prompts.
open_clip
PyTorch library for contrastive language-image pretraining.
LeRobot
PyTorch library for robot imitation learning and sim-to-real transfer.