[ORT][TRT] support FaceFusion #445
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
DefTruth
requested changes
Nov 12, 2024
| typedef _ONNXYOLOFaceNet YOLOV8Face; | ||
| typedef _ONNXFace_Swap InSwapper; | ||
| typedef _ONNXFace_Restoration GFPGAN; | ||
| typedef _ONNXFace_68Landmarks Face_68Landmarks; |
There was a problem hiding this comment.
这几个新增了类都属于FaceFusion系列，建议改成FaceFusion68Landmarks, FaceFusionRecognizer, FaceFusionSwap, FaceFusionRestoration, FaceFusionPipeline的风格。本仓库class名称采用驼峰法风格，不采用下划线。
wangzijian1010
changed the title
DefTruth
reviewed
Nov 15, 2024
| preprocessed_mat.convertTo(preprocessed_mat,CV_32FC3,1.0 / 255.f); | ||
| preprocessed_mat.convertTo(preprocessed_mat,CV_32FC3,1.0 / 1.f,0); | ||
|
|
||
| std::vector<float> model_martix = face_utils::load_npy("/home/facefusion-onnxrun/python/model_matrix.npy"); |
There was a problem hiding this comment.
我现在在测试大佬 我准备做完多线程的之后 再一次性全部修复 先不 merge
…switch of TensorRT
…CUDA version of NMS implementation.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request introduces new features and updates to the face recognition and manipulation functionalities in the project. It includes the addition of several new face-related models and their corresponding test files, as well as updates to the CMake configuration for these new models.
New Features:
face_68landmarks(lite/ort/cv/face_68landmarks.h,lite/ort/cv/face_68landmarks.cpp) [1] [2]face_recognizer(lite/ort/cv/face_recognizer.h,lite/ort/cv/face_recognizer.cpp)face_restoration(lite/ort/cv/face_restoration.h,lite/ort/cv/face_restoration.cpp)face_swap(lite/ort/cv/face_swap.h,lite/ort/cv/face_swap.cpp)face_fusion_pipeline(lite/ort/cv/face_fusion_pipeline.h,lite/ort/cv/face_fusion_pipeline.cpp) [1] [2]Updates to CMake Configuration:
OpenCV_LIBSvariable to includeopencv_calib3dincmake/opencv.cmake.examples/lite/CMakeLists.txt.Test Files:
test_lite_face_68landmarks.cpptest_lite_face_recognizer.cpptest_lite_face_restoration.cpptest_lite_face_swap.cpptest_lite_facefusion_pipeline.cppHeader File Updates:
lite/models.hto include the new face-related models and their typedefs. [1] [2] [3] [4]