SCUNet color real_psnr

Это LiteRT/TFLite-версия SCUNet от Binjia для blind real-image denoising цветных изображений. Модель конвертирована из официальной cszn/SCUNet (scunet_color_real_psnr.pth из KAIR v1.0, Apache-2.0). Вход и выход — float32 NHWC [1, 256, 256, 3] в sRGB [0, 1], файл весит около 47,2 МБ.

🔎 Ключевые особенности

• 📱 LiteRT/TFLite-формат для on-device инференса.
• 🖼 Blind real-image denoising: модель не требует отдельной карты шума.
• 🗂 Обработка перекрывающимися тайлами 256×256; глобальных операций нет.
• 🔄 Веса хранятся в FP16 с DEQUANTIZE, веса TRANSPOSE_CONV — FP32, активации FP32.
• ⚙ Маска shifted-window attention построена под размер тайла 256×256.
• ✅ Выход может слегка выходить за [0, 1] — значения нужно ограничивать.

🧭 Пример

Цветной кадр с шумом нарезается на перекрывающиеся тайлы 256×256, каждый прогоняется через scunet_color_real_psnr.tflite, результаты собираются обратно. Выходной sRGB при необходимости ограничивается диапазоном [0, 1].

Подробнее

Бенчмарки: отличие FP32-конверсии от PyTorch ~1e-6; TFLite с FP16-весами даёт минимум 75 dB PSNR относительно PyTorch на 54 тестовых тайлах; симуляция FP16 против FP32 — ≥69.7 dB.
Файл весит 47 200 352 байта, вход — [1, 256, 256, 3] float32 sRGB [0, 1].
Обрабатывайте изображение перекрывающимися тайлами 256×256; выход может слегка выходить за [0, 1], его нужно ограничить.
Вычисления: ~77 GMACs на тайл 256×256.