diff --git a/csrc/codebase/mmpose/CMakeLists.txt b/csrc/codebase/mmpose/CMakeLists.txt index 6d4c7dd562..ae58ce91ba 100644 --- a/csrc/codebase/mmpose/CMakeLists.txt +++ b/csrc/codebase/mmpose/CMakeLists.txt @@ -7,5 +7,6 @@ include(${CMAKE_SOURCE_DIR}/cmake/MMDeploy.cmake) file(GLOB_RECURSE SRCS ${CMAKE_CURRENT_SOURCE_DIR} "*.cpp") mmdeploy_add_module(${PROJECT_NAME} "${SRCS}") -target_link_libraries(${PROJECT_NAME} PRIVATE mmdeploy_opencv_utils) +target_link_libraries(${PROJECT_NAME} PRIVATE + mmdeploy::transform mmdeploy_opencv_utils) add_library(mmdeploy::mmpose ALIAS ${PROJECT_NAME}) diff --git a/csrc/codebase/mmpose/mmpose.h b/csrc/codebase/mmpose/mmpose.h index ed66f53a8e..a658d48947 100644 --- a/csrc/codebase/mmpose/mmpose.h +++ b/csrc/codebase/mmpose/mmpose.h @@ -3,6 +3,8 @@ #ifndef MMDEPLOY_MMPOSE_H #define MMDEPLOY_MMPOSE_H +#include + #include "codebase/common.h" #include "core/device.h" #include "core/module.h" diff --git a/demo/csrc/pose_detection.cpp b/demo/csrc/pose_detection.cpp index 14fa9c7391..253e965a8a 100644 --- a/demo/csrc/pose_detection.cpp +++ b/demo/csrc/pose_detection.cpp @@ -31,8 +31,7 @@ int main(int argc, char *argv[]) { mm_mat_t mat{img.data, img.rows, img.cols, 3, MM_BGR, MM_INT8}; mm_pose_detect_t *res{}; - int *res_count{}; - status = mmdeploy_pose_detector_apply(pose_estimator, &mat, 1, &res, &res_count); + status = mmdeploy_pose_detector_apply(pose_estimator, &mat, 1, &res); if (status != MM_SUCCESS) { fprintf(stderr, "failed to apply pose estimator, code: %d\n", (int)status); return 1;