fix pose demo and windows build (open-mmlab#307)

grimoire · Apr 15, 2022 · 2f3de01 · 2f3de01
1 parent c54d574
commit 2f3de01
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -3,6 +3,8 @@
 #ifndef MMDEPLOY_MMPOSE_H
 #define MMDEPLOY_MMPOSE_H
 
+#include <array>
+
 #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
@@ -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;