pycode19/nafnet-cascade-v3

NAFNet->RRDBNet Cascade v3 - Low-Light Denoising + 4x SR

Joint low-light denoising and 4x super-resolution (256x160 -> 1024x640). Pretrained NAFNet-SIDD-w32 denoiser cascaded into a RealESRNet-initialized RRDBNet-23, fine-tuned end-to-end on the IITM DLP NPPE3 dataset with a luminance-weighted Charbonnier loss and weight EMA. Leaderboard PSNR (grayscale, stride-8): 39.888.

import torch
from nafnet_cascade import CascadeModel

model = CascadeModel()
model.load_state_dict(torch.load("model_weights.pth", map_location="cpu"))
model.eval()
with torch.no_grad():
    sr = model(lr_image)  # 1x3xHxW float RGB in [0,1] -> 1x3x4Hx4W

Inference for the leaderboard score used x8 flip/rotation test-time augmentation.