Fennah01/retouch

Open Photo AI Models

This repository contains a collection of AI models used by Open Photo AI for photo enhancement tasks, including face detection, face recovery, image upscaling, denoising, sharpening, light adjustment, and color balance. Each model is designed to perform a specific function to improve the quality of photographs.

Models

Face Detection

• New York: based on RetinaFace (ResNet34), with fixed shape of 640x640, FP32 precision.

Face Recovery

• Athens: based on CodeFormer, with fixed shape of 512x512, FP32 and FP16 precisions.
• Santorini: based on GFPGAN, with fixed shape of 512x512, FP32 and FP16 precisions.

Upscale

• Tokyo: based on SwinIR, with fixed shape of 256x256, FP32 and FP16 precisions.
• Kyoto: based on Real-ESRGAN (General), with fixed shape of 256x256, FP32 and FP16 precisions.
• Saitama: based on Real-ESRGAN (Anime), with fixed shape of 256x256, FP32 and FP16 precisions.

Light Adjustment

• Paris: based on IAT with dynamic shapes, FP32 and FP16 precisions.

Color Balance

• Rio: based on Deep White-Balance with dynamic shapes, FP32 and FP16 precisions.

Denoise

• Stockholm: based on Restormer (Denoise) with fixed shape of 256x256, FP32 and FP16 precisions.
• Gothenburg: based on Restormer (Motion Deblur) with fixed shape of 256x256, FP32 and FP16 precisions.
• Malmo: based on Restormer (Focus Deblur) with fixed shape of 256x256, FP32 and FP16 precisions.
• Uppsala: based on Restormer (Derain) with fixed shape of 256x256, FP32 and FP16 precisions.

Sharpen

• Moscow: coming soon...