33 lines
632 B
C++
33 lines
632 B
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
#ifndef MMDEPLOY_MMPOSE_H
|
|
#define MMDEPLOY_MMPOSE_H
|
|
|
|
#include <array>
|
|
|
|
#include "codebase/common.h"
|
|
#include "core/device.h"
|
|
#include "core/module.h"
|
|
|
|
namespace mmdeploy {
|
|
namespace mmpose {
|
|
|
|
struct PoseDetectorOutput {
|
|
struct KeyPoint {
|
|
std::array<float, 2> bbox; // x, y
|
|
float score;
|
|
MMDEPLOY_ARCHIVE_MEMBERS(bbox, score);
|
|
};
|
|
std::vector<KeyPoint> key_points;
|
|
MMDEPLOY_ARCHIVE_MEMBERS(key_points);
|
|
};
|
|
|
|
DECLARE_CODEBASE(MMPose, mmpose);
|
|
|
|
} // namespace mmpose
|
|
|
|
MMDEPLOY_DECLARE_REGISTRY(mmpose::MMPose);
|
|
} // namespace mmdeploy
|
|
|
|
#endif // MMDEPLOY_MMPOSE_H
|