mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
33 lines
692 B
C
33 lines
692 B
C
|
// Copyright (c) OpenMMLab. All rights reserved.
|
||
|
|
|
||
|
|
#ifndef MMDEPLOY_MMROTATE_H
|
||
|
|
#define MMDEPLOY_MMROTATE_H
|
||
|
|
|
||
|
|
#include <array>
|
||
|
|
|
||
|
|
#include "codebase/common.h"
|
||
|
|
#include "core/device.h"
|
||
|
|
#include "core/module.h"
|
||
|
|
|
||
|
|
namespace mmdeploy {
|
||
|
|
namespace mmrotate {
|
||
|
|
|
||
|
|
struct RotatedDetectorOutput {
|
||
|
|
struct Detection {
|
||
|
|
int label_id;
|
||
|
|
float score;
|
||
|
|
std::array<float, 5> rbbox; // cx,cy,w,h,ag
|
||
|
|
MMDEPLOY_ARCHIVE_MEMBERS(label_id, score, rbbox);
|
||
|
|
};
|
||
|
|
std::vector<Detection> detections;
|
||
|
|
MMDEPLOY_ARCHIVE_MEMBERS(detections);
|
||
|
|
};
|
||
|
|
|
||
|
|
DECLARE_CODEBASE(MMRotate, mmrotate);
|
||
|
|
} // namespace mmrotate
|
||
|
|
|
||
|
|
MMDEPLOY_DECLARE_REGISTRY(mmrotate::MMRotate);
|
||
|
|
} // namespace mmdeploy
|
||
|
|
|
||
|
|
#endif // MMDEPLOY_MMROTATE_H
|