MBbox.h

Go to the documentation of this file.

#pragma once
 
#include <vector>
#include <string>
#include "common/macro.h"
 
namespace mmind {
namespace dl {
 
struct MPointF
{
    float x;
    float y;
 
    constexpr MPointF() : x(0.0), y(0) {}
 
    constexpr MPointF(float x, float y) : x(x), y(y) {}
 
    MPointF& operator+=(const MPointF& other) noexcept
    {
        x += other.x;
        y += other.y;
        return *this;
    }
};
 
constexpr MPointF operator+(const MPointF& p1, const MPointF& p2)
{
    return {p1.x + p2.x, p1.y + p2.y};
}
 
struct MBbox
{
    MBbox() = default;
    MBbox(float startX, float startY, float endX, float endY, float confidence, int labelIndex,
          const std::string& labelName, bool recoverToFixedSize = false,
          const std::vector<float>& confidenceSeq = {})
        : startX(startX), startY(startY), endX(endX), endY(endY),
          recoverToFixedSize(recoverToFixedSize)
    {
        extraInfo.confidence = confidence;
        extraInfo.labelIndex = labelIndex;
        extraInfo.labelName = labelName;
        extraInfo.confidenceSeq = confidenceSeq;
    }
    float width() const { return endX - startX; }
    float height() const { return endY - startY; }
    float area() const { return width() * height(); }
    std::vector<float> centerPoint() const { return {(endX + startX) / 2, (endY + startY) / 2}; }
 
    float startX{0.0};
    float startY{0.0};
    float endX{0.0};
    float endY{0.0};
 
    bool recoverToFixedSize{false};
 
    struct ExtraInfo
    {
        float confidence{};
        int labelIndex;
        std::string labelName{};
        std::vector<float> confidenceSeq{};
    } extraInfo;
};

struct MMIND_DL_SDK_CC_API MRotatedBbox
{
    MRotatedBbox() = default;
    MRotatedBbox(const MPointF& center, float width, float height, float angle, float confidence,
                 int labelIndex, const std::string& labelName);
 
    MRotatedBbox(const MBbox& bbox);
 
    std::vector<MPointF> toBboxPoints() const;
 
    // only support bbox with no angle
    MBbox toBbox() const;
 
    MPointF center{0.0, 0.0};
    float width{0.0};
    float height{0.0};
    float angle{0.0};
 
    struct ExtraInfo
    {
        float confidence{};
        int labelIndex;
        std::string labelName{};
    } extraInfo;
};
} // namespace dl
} // namespace mmind