c++ vector 添加结构体元素,检测结果坐标的几个写法
短信预约 -IT技能 免费直播动态提醒
1、用向量vector ,添加的时候判断,相同名字的元素,置信度最大的框进行保存,如果有了就替换相关值,或者全部push 后遍历一遍删除相同的框子中,置信度最低的那个框
// 检测 //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、用字典,添加自定义结构体变量框数据
// 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); //xywhfloat 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 写法
//当你想要对自定义的结构体进行排序时,你可以通过重载operator<来定义排序的规则。//以下是一个示例,展示了如何在结构体中重载operator<来按照置信度(prob)进行排序:cppstruct 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向量进行排序,如下所示:cppstd::vector<Detect> dets;// 添加元素到dets...std::sort(dets.begin(), dets.end());这将使用重载的operator<来对dets中的元素进行排序,按照置信度从低到高的顺序进行排列。希望这对你有帮助!👍🏻
来源地址:https://blog.csdn.net/m0_37192554/article/details/133763800
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341