resenh

resenh — веса для библиотеки устойчивого восстановления и улучшения изображений при слабом освещении от SharadhNaiduTrains / Team V-Max (T46). Внутри Retinexformer на 1.6M параметров и адаптивный классический пайплайн (Retinex, CLAHE, edge-aware denoising); контроллер качества смешивает две ветки для каждого фото отдельно.

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

• 🔥 Гибридный режим: классическая ветка и Retinexformer работают параллельно, quality controller смешивает результат по каждому изображению.
• 👀 В комплекте несколько чекпоинтов: general для LOL + LSRW, отдельные fine-tuned для LOL-v1/v2, официальные Retinexformer и ONNX для браузера.
• 💪 Пакет проверяет SHA-256 каждого файла и отказывается загружать веса при несовпадении.
• 🐱 Есть браузерное демо на ONNX Runtime Web с WebGPU; 1 MP фото в ноутбучном браузере гибрид обрабатывает примерно за 6 с.
• 💸 MIT licence; веса Retinexformer — MIT, авторы Cai et al., ICCV 2023.

💪 Пример

import resenh

res = resenh.enhance("night.jpg")
res.save("night_restored.png")
print(res.summary())

Подробнее

PSNR: гибрид 25.13 dB на LOL-v1, 22.82 dB на LOL-v2-Real, 19.26 dB на LSRW test.
На LOL-v2-Real SSIM 0.840, LPIPS 0.171, NIQE 3.59 — как у сети без классической ветки.
Скорость: 1 MP — 0.49 с только сеть и 1.9 с полный гибрид на RTX 4070 Laptop; в браузере с WebGPU около 6 с.
Отдельные fine-tuned чекпоинты дают SSIM/LPIPS 0.852/0.139 на LOL-v2-Real, но PSNR ниже — 20.82 dB.
PSNR считается по сохранённому 8-битному изображению, без коррекции яркости по ground truth и выбора чекпоинта на тесте.