Skip to content

Commit 2f3de01

Browse files
irexycgrimoire
authored andcommitted
fix pose demo and windows build (open-mmlab#307)
1 parent c54d574 commit 2f3de01

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

csrc/codebase/mmpose/CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,5 +7,6 @@ include(${CMAKE_SOURCE_DIR}/cmake/MMDeploy.cmake)
77

88
file(GLOB_RECURSE SRCS ${CMAKE_CURRENT_SOURCE_DIR} "*.cpp")
99
mmdeploy_add_module(${PROJECT_NAME} "${SRCS}")
10-
target_link_libraries(${PROJECT_NAME} PRIVATE mmdeploy_opencv_utils)
10+
target_link_libraries(${PROJECT_NAME} PRIVATE
11+
mmdeploy::transform mmdeploy_opencv_utils)
1112
add_library(mmdeploy::mmpose ALIAS ${PROJECT_NAME})

csrc/codebase/mmpose/mmpose.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
#ifndef MMDEPLOY_MMPOSE_H
44
#define MMDEPLOY_MMPOSE_H
55

6+
#include <array>
7+
68
#include "codebase/common.h"
79
#include "core/device.h"
810
#include "core/module.h"

demo/csrc/pose_detection.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,7 @@ int main(int argc, char *argv[]) {
3131
mm_mat_t mat{img.data, img.rows, img.cols, 3, MM_BGR, MM_INT8};
3232

3333
mm_pose_detect_t *res{};
34-
int *res_count{};
35-
status = mmdeploy_pose_detector_apply(pose_estimator, &mat, 1, &res, &res_count);
34+
status = mmdeploy_pose_detector_apply(pose_estimator, &mat, 1, &res);
3635
if (status != MM_SUCCESS) {
3736
fprintf(stderr, "failed to apply pose estimator, code: %d\n", (int)status);
3837
return 1;

0 commit comments

Comments
 (0)