[Bugfix] Fix Relative Path (#448)

Author: wangzijian1010

CMakeLists.txt

@@ -26,6 +26,8 @@ set(CMAKE_CUDA_ARCHITECTURES 89) # For RTX 20xx series
 # 启用 CUDA
 enable_language(CUDA)
 set(LITE_AI_ROOT_DIR ${CMAKE_SOURCE_DIR})
+# 将 CMAKE_SOURCE_DIR 传递给代码
+add_definitions(-DSOURCE_PATH="${CMAKE_SOURCE_DIR}")
 
 option(ENABLE_TEST "build test examples." OFF)
 option(ENABLE_DEBUG_STRING "enable DEBUG string or not" ON)

examples/lite/resources/model_matrix.npy

@@ -19,7 +19,9 @@ void Face_Swap::preprocess(cv::Mat &target_face, std::vector<float> source_image
     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");
+    // 使用 CMake 传递的 SOURCE_PATH 宏
+    std::string model_matrix_path = std::string(SOURCE_PATH) + "/examples/lite/resources/model_matrix.npy";
+    std::vector<float> model_martix = face_utils::load_npy(model_matrix_path);
 
     processed_source_embeding= face_utils::dot_product(source_image_embeding,model_martix,512);
 

lite/ort/cv/face_swap.cpp

@@ -17,8 +17,9 @@ void TRTFaceFusionFaceSwap::preprocess(cv::Mat &target_face, std::vector<float>
     cv::cvtColor(preprocessed_mat,preprocessed_mat,cv::COLOR_BGR2RGB);
     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");
+    // 使用 CMake 传递的 SOURCE_PATH 宏
+    std::string model_matrix_path = std::string(SOURCE_PATH) + "/examples/lite/resources/model_matrix.npy";
+    std::vector<float> model_martix = face_utils::load_npy(model_matrix_path);
 
     processed_source_embeding= face_utils::dot_product(source_image_embeding,model_martix,512);