CodeFormer
Code
ورقة | صفحة المشروع | فيديو
Shangchen Zhou ، Kelvin CK Chan ، Chongyi Li ، Chen Change Loy
S-Lab ، جامعة Nanyang التكنولوجية
إذا كان CodeFormer مفيدًا لصورك أو مشاريعك ، فيرجى مساعدة هذا الريبو. شكرًا! ؟
dlib كخيار جديد للكشف عن الوجه ، فإنه ينتج هوية وجه أكثر دقة.--input_path [YOUR_VIDEO.mp4] . جربها لتعزيز مقاطع الفيديو الخاصة بك! ؟
requirements.txt # git clone this repository
git clone https://github.com/sczhou/CodeFormer
cd CodeFormer
# create new anaconda env
conda create -n codeformer python=3.8 -y
conda activate codeformer
# install python dependencies
pip3 install -r requirements.txt
python basicsr/setup.py develop
conda install -c conda-forge dlib (only for face detection or cropping with dlib)
قم بتنزيل نماذج Facelib و Dlib PretRained من [الإصدارات | محرك Google | OneDrive] إلى المجلد weights/facelib . يمكنك تنزيل النماذج المسبقة يدويًا أو التنزيل عن طريق تشغيل الأمر التالي:
python scripts/download_pretrained_models.py facelib
python scripts/download_pretrained_models.py dlib (only for dlib face detector)
قم بتنزيل النماذج المسبقة CodeFormer من [الإصدارات | محرك Google | OneDrive] إلى المجلد weights/CodeFormer . يمكنك تنزيل النماذج المسبقة يدويًا أو التنزيل عن طريق تشغيل الأمر التالي:
python scripts/download_pretrained_models.py CodeFormer
يمكنك وضع صور الاختبار في مجلد inputs/TestWhole . إذا كنت ترغب في اختبار الوجوه المزروعة والمحاذاة ، فيمكنك وضعها في مجلد inputs/cropped_faces . يمكنك الحصول على الوجوه المزروعة والمحاذاة عن طريق تشغيل الأمر التالي:
# you may need to install dlib via: conda install -c conda-forge dlib
python scripts/crop_align_face.py -i [input folder] -o [output folder]
[ملاحظة] إذا كنت ترغب في مقارنة codeformer في ورقتك ، فيرجى تشغيل الأمر التالي الذي يشير --has_aligned (للوجه المقطوع والمحاذاة) ، حيث أن الأمر الخاص بالصورة بأكملها سوف يتضمن عملية الانصهار الخلفي للوجه التي قد تضر بملمس الشعر على الحدود ، مما يؤدي إلى مقارنة غير مبال.
وزن الإخلاص W يضع في [0 ، 1]. بشكل عام ، يميل W الأصغر إلى إنتاج نتيجة عالية الجودة ، في حين أن W الأكبر يعطي نتيجة أعلى. سيتم حفظ النتائج في مجلد results .
؟ استعادة الوجه (وجه اقتصاص ومحاذاة)
# For cropped and aligned faces (512x512)
python inference_codeformer.py -w 0.5 --has_aligned --input_path [image folder]|[image path]
️ تحسين الصورة
# For whole image
# Add '--bg_upsampler realesrgan' to enhance the background regions with Real-ESRGAN
# Add '--face_upsample' to further upsample restorated face with Real-ESRGAN
python inference_codeformer.py -w 0.7 --input_path [image folder]|[image path]
؟ تعزيز الفيديو
# For Windows/Mac users, please install ffmpeg first
conda install -c conda-forge ffmpeg
# For video clips
# Video path should end with '.mp4'|'.mov'|'.avi'
python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0 --input_path [video path]
؟ تلوين الوجه (وجه اقتصاص ومحاذاة)
# For cropped and aligned faces (512x512)
# Colorize black and white or faded photo
python inference_colorization.py --input_path [image folder]|[image path]
؟ وجه الوجه (الوجه المقطوع والمحاذاة)
# For cropped and aligned faces (512x512)
# Inputs could be masked by white brush using an image editing app (e.g., Photoshop)
# (check out the examples in inputs/masked_faces)
python inference_inpainting.py --input_path [image folder]|[image path]
يمكن العثور على أوامر التدريب في المستندات: اللغة الإنجليزية |简体中文.
إذا كان عملنا مفيدًا لبحثك ، فيرجى التفكير في:
@inproceedings{zhou2022codeformer,
author = {Zhou, Shangchen and Chan, Kelvin C.K. and Li, Chongyi and Loy, Chen Change},
title = {Towards Robust Blind Face Restoration with Codebook Lookup TransFormer},
booktitle = {NeurIPS},
year = {2022}
}
تم ترخيص هذا المشروع بموجب ترخيص NTU S-LAB 1.0. يجب أن تتبع إعادة التوزيع والاستخدام هذا الترخيص.
يعتمد هذا المشروع على الأساسيات. يتم إحضار بعض الرموز من العنان للمحولات و yolov5-face و facexlib. نعتمد أيضًا real-esrgan لدعم تحسين صورة الخلفية. شكرا لأعمالهم الرائعة.
إذا كان لديك أي أسئلة ، فلا تتردد في الوصول إلي على [email protected] .
