Python, most likely any 3.x would do, but I'm currently running it on 3.10.8
python -m venv venv
venv\Scripts\activate.bat
pip install -r requirements.txtpython -m venv venv
source venv/bin/activate
pip install -r requirements.txtpython main.py --resolution=RESOLUTION --source-path=SOURCE_PATH --target-path=TARGET_PATHRESOLUTIONis the cropping resolution (usually one of the following:512,768,1024) at the moment only square resolution is supported, default is512SOURCE_PATHis the path where the original images are fetched from, you should have subfolder for each person for example:there should be at least one subfolder for the script to do something, but if there are more, all will be croppedSOURCE_PATH ritaora billieeilish henrycavill someoneelse
TARGET_PATHis the path where the outputs will be savedYou only need to make sure that theTARGET_PATH cropped ritaora billieeilish henrycavill someoneelseTARGET_PATHexists, other subfolders will be created automatically
- If the script is running slow, check if the
DLIBlibrary is usingCUDA:If the result is false then you have to compile the DLIB library for CUDA: https://www.linkedin.com/pulse/installing-dlib-cuda-support-windows-10-chitransh-mundrapython main.py --check-dlib
- If you get some errors -> please report them to me, I will try to do my best to help with them / improve the code
- Tested with
jpg,pngandwebpformats but most likely others should work too - For best experience, try using images with only one person in the photo, there is however some code to recognize which person is our target but this code is WIP still
- Currently only square resolutions are available (this is why you only provide one value)
- If you want to crop to higher res (768 or 1024) then smaller images will get upscaled
- There is no up-scaling for 512 images though
- Technically you can crop to any resolution, but I have not tested other resolutions besides the 3 supported ones
- If something is wrong with an image and I recognize it (wrong format, too big, etc.) I will skip it and proceed to the next, you can check the output for
skipped filesif you are interested in the files that were problematic
- https://stackoverflow.com/questions/51943050/dlib-not-using-cuda
- https://www.geeksforgeeks.org/how-to-install-dlib-library-for-python-in-windows-10/
- ageitgey/face_recognition#1143 (check comment about cpu)
- https://www.linkedin.com/pulse/installing-dlib-cuda-support-windows-10-chitransh-mundra
- https://discourse.cmake.org/t/cmake-issues-in-creation-of-dlib-for-gpu-in-windows/6260
- CIVITAI profile: http://civitai.com/user/malcolmrey
- BuyMeACoffee: https://www.buymeacoffee.com/malcolmrey