c++ vector 添加结构体元素,检测结果坐标的几个写法

1、用向量vector ,添加的时候判断,相同名字的元素,置信度最大的框进行保存,如果有了就替换相关值,或者全部push 后遍历一遍删除相同的框子中,置信度最低的那个框

cpp 复制代码
// 检测 //
struct Detect
{
    std::string name;
    float prob;
    cv::Rect rect;
};


void add_det(std::vector<Detect>& dets)
// 输入引用类型的 数据方便修改
// 在每次添加的时候遍历判断一下,如果没有就添加,如果有,就比较判断更新最大的那个元素
{
    bool found = false;
    for (auto& det : dets) {
        if (det.name == label) {
            if (conf > det.prob) {
                det.prob = conf;
                det.rect = bbox;
            }
            found = true;
            break;
        }
    }

    if (!found) {
        dets.push_back({label, conf, bbox});
    }
 }



/
	std::vector<Detect> dets;

    std::string label = "狗";
    cv::Rect bbox(363, 388, 69, 57); //xywh
    float conf = 0.797;
    dets.push_back({label, conf, bbox});

    bbox = cv::Rect(366, 209, 101, 91); //xywh
    dets.push_back({"猫", 0.702, bbox});

    bbox = cv::Rect(113, 94, 533, 453); //xywh
    dets.push_back({"猫", 0.669, bbox});

2、用字典,添加自定义结构体变量框数据

cpp 复制代码
// std::map中的一个键值对。键值对由键和值组成,通过it->first可以访问键,通过it->second可以访问值
struct Detect
{
    std::string name;
    float prob;
    cv::Rect rect;
};
//


std::map<std::string, Detect> detectMap;

std::string label = "猫";
cv::Rect bbox(363, 388, 69, 57); //xywh
float conf = 0.797;

// 查找是否已存在相同标签的元素
auto it = detectMap.find(label);
if (it != detectMap.end()) {
    // 如果已存在,比较置信度并更新
    if (conf > it->second.prob) {
        it->second.prob = conf;
        it->second.rect = bbox;
    }
} else {
    // 如果不存在,添加新元素
    detectMap.insert({label, {label, conf, bbox}});
}

3、 重载操作符 opreate 写法

cpp 复制代码
//当你想要对自定义的结构体进行排序时,你可以通过重载operator<来定义排序的规则。
//以下是一个示例,展示了如何在结构体中重载operator<来按照置信度(prob)进行排序:

cpp
struct Detect {
    std::string name;
    float prob;
    cv::Rect rect;

    bool operator<(const Detect& other) const {
        return prob < other.prob;
    }
};

//在这个示例中,我们将operator<重载为比较两个Detect结构体的置信度(prob)大小。根据你的需求,你也可以选择比较其他的字段。

// 然后,你可以使用标准库中的排序算法(例如std::sort)对dets向量进行排序,如下所示:

cpp
std::vector<Detect> dets;

// 添加元素到dets...

std::sort(dets.begin(), dets.end());
这将使用重载的operator<来对dets中的元素进行排序,按照置信度从低到高的顺序进行排列。

希望这对你有帮助!👍🏻
相关推荐
大模型码小白1 小时前
【Python零基础教程】继承、多态与魔法函数:面向对象编程三大核心特性详解
java·大数据·开发语言·人工智能·python·ai编程
_wyt0012 小时前
多重背包问题详解
c++·背包dp
段一凡-华北理工大学3 小时前
向量数据库实战:选型、调优与落地~系列文章12:文本分块策略实战:chunk_size 怎么选?重叠多少?
开发语言·数据库·后端·oracle·rust·工业智能体·高炉智能化
Ljwuhe3 小时前
C++——多态
开发语言·c++
越甲八千4 小时前
STL stack为何没有迭代器
c++
心平气和量大福大4 小时前
C#-WPF-Window主窗体
开发语言·c#·wpf
从零开始的代码生活_5 小时前
C++ 继承详解:访问控制、对象模型、菱形继承与设计取舍
开发语言·c++·后端·学习·算法
云小逸5 小时前
【C++ 第七阶段:模板、泛型编程与工程综合详解】
开发语言·c++
GIS阵地6 小时前
QgsSingleBandPseudoColorRenderer 完整详解(QGIS 3.40.13 C++)
开发语言·前端·c++·qt·qgis
王维同学6 小时前
[原创][Windows C++]LSA 认证、安全与通知包的注册表枚举
c++·windows·安全