Skip to content

Support inference on MacOS with MPS or CPU and add face detection script#188

Open
walkoncross wants to merge 14 commits into
sicxu:masterfrom
walkoncross:master
Open

Support inference on MacOS with MPS or CPU and add face detection script#188
walkoncross wants to merge 14 commits into
sicxu:masterfrom
walkoncross:master

Conversation

@walkoncross
Copy link
Copy Markdown

@walkoncross walkoncross commented Jan 22, 2025

  • inference on MacOS with MPS:
# get reconstruction results of example images
python test.py --name=<model_name> --epoch=20 --img_folder=./datasets/examples --device mps --renderer_type face3d

# no visualization
python test.py --name=<model_name> --epoch=20 --img_folder=./datasets/examples --device mps --renderer_type none --no_viz
  • inference on MacOS with CPU:
# get reconstruction results of example images
python test.py --name=<model_name> --epoch=20 --img_folder=./datasets/examples --device cpu --renderer_type face3d

python test.py --name=<model_name> --epoch=20 --img_folder=./datasets/examples --device cpu --renderer_type none --no_viz
  • detect faces and landmarks:
python detect_faces_by_insightface.py <folder_to_test_images>

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant