Port GPEN to modelzoo / coreml

PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

https://qiita.com/PINTO

MIT License

3.62k stars 576 forks source link

Port GPEN to modelzoo / coreml #291

Open yousifa opened 2 years ago

yousifa commented 2 years ago

Issue Type

Feature Request

OS

Mac OS

OS architecture

aarch64

Programming Language

Other

Framework

CoreML

Model name and Weights/Checkpoints URL

https://github.com/yangxy/GPEN

Description

Would be awesome to port GPEN to modelzoo for face restoration in coreml. It provides great results.

Relevant Log Output

No response

URL or source code for simple inference testing code

No response

PINTO0309 commented 2 years ago

License
© Alibaba, 2021. For academic and non-commercial use only.

PINTO0309 commented 2 years ago

This model. Real-ESRGAN + RetinaFace + FaceGAN

Real-ESRGAN - BSD 3-Clause License https://github.com/PINTO0309/PINTO_model_zoo/tree/main/133_Real-ESRGAN
RetinaFace - MIT License https://github.com/PINTO0309/PINTO_model_zoo/tree/main/096_RetinaFace https://github.com/PINTO0309/Face_Mask_Augmentation
FaceGAN - Alibaba, 2021. For academic and non-commercial use only. Currently under analysis.