Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Update] FaceFusion Speed up #446

Merged
merged 19 commits into from
Dec 9, 2024
Merged

Conversation

wangzijian1010
Copy link
Collaborator

This pull request includes several changes to improve the functionality and performance of the lite module, particularly focusing on face recognition and restoration. The most important changes include adjustments to the build configuration, updates to file paths, enhancements to the face detection pipeline, and optimizations for multi-threading and performance measurement.

Build Configuration Changes:

  • Disabled ENABLE_TEST and ENABLE_TENSORRT options in CMakeLists.txt to streamline the build process.

File Path Updates:

  • Updated file paths in examples/lite/cv/test_lite_facefusion_pipeline.cpp to use absolute paths for consistency and reliability. [1] [2]

Face Detection and Recognition Enhancements:

  • Added a new function test_tensorrt in examples/lite/cv/test_lite_facefusion_pipeline.cpp to support TensorRT engine for face detection.
  • Introduced multi-threading optimizations in lite/trt/cv/trt_face_68landmarks_mt.cpp by resizing vectors and modifying function signatures to include thread IDs. [1] [2] [3] [4] [5] [6] [7] [8]

Performance Measurement:

  • Added performance measurement code to various functions in lite/trt/cv/trt_face_restoration.cpp and lite/trt/cv/trt_face_swap.cpp to log preprocessing, inference, and postprocessing times. [1] [2] [3] [4] [5]

Miscellaneous:

  • Changed the author name in the header comment of lite/ort/cv/yolofacev8.cpp.
  • Added comments for clarity in lite/ort/cv/yolofacev8.cpp.
  • Included additional header files for postprocessing in lite/trt/cv/trt_face_recognizer.h and lite/trt/cv/trt_face_restoration.h. [1] [2]

Copy link
Owner

@DefTruth DefTruth left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@wangzijian1010 wangzijian1010 merged commit 5f4938f into DefTruth:main Dec 9, 2024
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.

2 participants