1
0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Yue Zhou 9411fe0466
Add nms_rotated ort op ()
* fix pose demo and windows build ()

* init

* Update nms_rotated.cpp

* add postprocessing_masks gpu version ()

* add postprocessing_masks gpu version

* default device cpu

* pre-commit fix

Co-authored-by: hadoop-basecv <hadoop-basecv@set-gh-basecv-serving-classify11.mt>

* fixed a bug causes text-recognizer to fail when (non-NULL) empty bboxes list is passed ()

* [Fix] include missing <type_traits> for formatter.h ()

* fix formatter

* relax GCC version requirement

* fix

* fix lint

* fix lint

* [Fix] MMEditing cannot save results when testing ()

* fix show

* lint

* remove redundant codes

* resolve comment

* type hint

* docs(build): fix typo ()

* docs(build): add missing build option

* docs(build): add onnx install

* style(doc): trim whitespace

* docs(build): revert install onnx

* docs(build): add ncnn LD_LIBRARY_PATH

* docs(build): fix path error

* fix openvino export tmp model, add binary flag ()

* init circleci ()

* fix wrong input mat type ()

* fix wrong input mat type

* fix lint

* fix(docs): remove redundant doc tree ()

* fix missing ncnn_DIR & InferenceEngine_DIR ()

* update doc

Co-authored-by: Chen Xin <xinchen.tju@gmail.com>
Co-authored-by: Shengxi Li <982783556@qq.com>
Co-authored-by: hadoop-basecv <hadoop-basecv@set-gh-basecv-serving-classify11.mt>
Co-authored-by: lzhangzz <lzhang329@gmail.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: tpoisonooo <khj.application@aliyun.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
2022-04-18 17:14:47 +08:00

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