J1gA/zernike-phase-diversity-demo
0
Phase-diversity CNN — интерфейс инференса
Демонстрация двух моделей восстановления 14 коэффициентов разложения Цернике (OSA-нумерация, радианы) из пары PSF (in-focus + defocused).
Модели
Веса подтягиваются при старте контейнера через huggingface_hub.hf_hub_download и кэшируются. Холодный старт — около 10–30 сек.
Архитектура
EfficientNet-B2 (PyTorch + timm), вход — два изображения 256×256 в одном двухканальном тензоре, выход — вектор размерности 14 (Z₁…Z₁₄ в радианах).
Использование
- Выбрать модель в выпадающем списке.
- Либо подгрузить пару изображений PSF (NPZ / TIFF / PNG), либо использовать встроенный генератор для синтетической проверки.
- Получить предсказанные коэффициенты + WF RMS / Strehl метрики коррекции.
