YOLOv5 | NMS | 手动实现NMS

2024-04-06 14:44
文章标签 实现 yolov5 手动 nms

本文主要是介绍YOLOv5 | NMS | 手动实现NMS,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

这个NMS可以按照相同的类别进行nms,不同类别是不会进行nms的,可以加一个“开关”(参数)来控制是否使用进行同类别iou,但是懒得加了哈哈哈哈 ,就是一个if语句,offset这个简直绝了,学的源码的哈哈哈哈哈哈,不信你去看

YOLOv5 | NMS | 源码解析-CSDN博客

import numpy as np
import torchdef nms(boxes, score, conf_thres, iou_thres, classes):tf = score[:] > conf_thresscore = score[tf]keep = []idx = np.argsort(score)[::-1]pred = torch.hstack((boxes,classes))offset = pred[:,4:5] * 4096boxes = pred[:,0:4] + offset# print(boxes)while len(idx) > 0:keep.append(idx[0])overlap = np.zeros_like(idx[1:], dtype= np.float32)for i, j in enumerate(idx[1:]):bbox1 = boxes[idx[0]]bbox2 = boxes[j]out = iou(bbox1, bbox2)overlap[i] = outidx = idx[1:][overlap < iou_thres]return keep# box = []# for i in keep:#     box.append(boxes[i])# return boxdef iou(bbox1,bbox2):x1, y1, w1, h1 = bbox1 # left up right bottomx2, y2, w2, h2 = bbox2if (x1 <= ((x2+ w2)/2) <= w1) or ((y1 <= ((y2+ h2)/2) <= h1)): # 必须加上等号left_top_x = max(x1, x2)left_top_y = max(y1, y2) right_bottom_x = min(w1, w2)right_bottom_y = min(h1, h2)I = (right_bottom_x - left_top_x) * (right_bottom_y - left_top_y)o = (w1 - x1) * (h1- y1) + (w2 - x2) * (h2 - y2) - IIoU = I / oreturn IoUelse: # 不相交return 0if __name__ == '__main__':boxes = torch.Tensor([[100, 100, 200, 200],[120, 110, 220, 210],[300, 320, 400, 400],[180, 100, 300, 180]])scores = np.array([0.9, 0.8, 0.7, 0.6])classes = torch.Tensor([[0],[1],[1],[2]])# np.concatenate 先转置 可以拼接 out = nms(boxes, scores, 0.5, 0.25,classes)print(out)

写的比较简单,不会高级的写法

这篇关于YOLOv5 | NMS | 手动实现NMS的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/880118

相关文章

mybatis执行insert返回id实现详解

《mybatis执行insert返回id实现详解》MyBatis插入操作默认返回受影响行数,需通过useGeneratedKeys+keyProperty或selectKey获取主键ID,确保主键为自... 目录 两种方式获取自增 ID:1. ​​useGeneratedKeys+keyProperty(推

Spring Boot集成Druid实现数据源管理与监控的详细步骤

《SpringBoot集成Druid实现数据源管理与监控的详细步骤》本文介绍如何在SpringBoot项目中集成Druid数据库连接池,包括环境搭建、Maven依赖配置、SpringBoot配置文件... 目录1. 引言1.1 环境准备1.2 Druid介绍2. 配置Druid连接池3. 查看Druid监控

Linux在线解压jar包的实现方式

《Linux在线解压jar包的实现方式》:本文主要介绍Linux在线解压jar包的实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录linux在线解压jar包解压 jar包的步骤总结Linux在线解压jar包在 Centos 中解压 jar 包可以使用 u

c++ 类成员变量默认初始值的实现

《c++类成员变量默认初始值的实现》本文主要介绍了c++类成员变量默认初始值,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧... 目录C++类成员变量初始化c++类的变量的初始化在C++中,如果使用类成员变量时未给定其初始值,那么它将被

Qt使用QSqlDatabase连接MySQL实现增删改查功能

《Qt使用QSqlDatabase连接MySQL实现增删改查功能》这篇文章主要为大家详细介绍了Qt如何使用QSqlDatabase连接MySQL实现增删改查功能,文中的示例代码讲解详细,感兴趣的小伙伴... 目录一、创建数据表二、连接mysql数据库三、封装成一个完整的轻量级 ORM 风格类3.1 表结构

基于Python实现一个图片拆分工具

《基于Python实现一个图片拆分工具》这篇文章主要为大家详细介绍了如何基于Python实现一个图片拆分工具,可以根据需要的行数和列数进行拆分,感兴趣的小伙伴可以跟随小编一起学习一下... 简单介绍先自己选择输入的图片,默认是输出到项目文件夹中,可以自己选择其他的文件夹,选择需要拆分的行数和列数,可以通过

Python中将嵌套列表扁平化的多种实现方法

《Python中将嵌套列表扁平化的多种实现方法》在Python编程中,我们常常会遇到需要将嵌套列表(即列表中包含列表)转换为一个一维的扁平列表的需求,本文将给大家介绍了多种实现这一目标的方法,需要的朋... 目录python中将嵌套列表扁平化的方法技术背景实现步骤1. 使用嵌套列表推导式2. 使用itert

Python使用pip工具实现包自动更新的多种方法

《Python使用pip工具实现包自动更新的多种方法》本文深入探讨了使用Python的pip工具实现包自动更新的各种方法和技术,我们将从基础概念开始,逐步介绍手动更新方法、自动化脚本编写、结合CI/C... 目录1. 背景介绍1.1 目的和范围1.2 预期读者1.3 文档结构概述1.4 术语表1.4.1 核

在Linux中改变echo输出颜色的实现方法

《在Linux中改变echo输出颜色的实现方法》在Linux系统的命令行环境下,为了使输出信息更加清晰、突出,便于用户快速识别和区分不同类型的信息,常常需要改变echo命令的输出颜色,所以本文给大家介... 目python录在linux中改变echo输出颜色的方法技术背景实现步骤使用ANSI转义码使用tpu

Python使用python-can实现合并BLF文件

《Python使用python-can实现合并BLF文件》python-can库是Python生态中专注于CAN总线通信与数据处理的强大工具,本文将使用python-can为BLF文件合并提供高效灵活... 目录一、python-can 库:CAN 数据处理的利器二、BLF 文件合并核心代码解析1. 基础合