lzhangzz a494a6f6ff
[SDK] sync changes according to performance benchmarks (#297)
* sync SDK changes according to performance benchmarks

* fix end-of-file lint

* fix clang-format issue

* fix clang-format by adding 'clang-format off'

* remove useless casts

* remove 'data' argument of 'operator()'

* change 'Tensor2Img' to 'TensorToImg' according to spec

* correct tensor's name according spec

Co-authored-by: lvhan028 <lvhan_028@163.com>
2021-12-16 13:51:22 +08:00

32 lines
745 B
C++

// Copyright (c) OpenMMLab. All rights reserved.
#ifndef MMDEPLOY_SRC_CODEBASE_MMDET_MMDET_H_
#define MMDEPLOY_SRC_CODEBASE_MMDET_MMDET_H_
#include "codebase/common.h"
#include "core/device.h"
#include "core/mat.h"
#include "core/module.h"
#include "core/serialization.h"
namespace mmdeploy::mmdet {
struct DetectorOutput {
struct Detection {
int index;
int label_id;
float score;
std::array<float, 4> bbox; // left, top, right, bottom
Mat mask;
MMDEPLOY_ARCHIVE_MEMBERS(index, label_id, score, bbox, mask);
};
std::vector<Detection> detections;
MMDEPLOY_ARCHIVE_MEMBERS(detections);
};
DECLARE_CODEBASE(MMDetection, mmdet);
} // namespace mmdeploy::mmdet
#endif // MMDEPLOY_SRC_CODEBASE_MMDET_MMDET_H_