yolov5 后处理
时间: 2023-08-24 07:09:35 浏览: 67
Yolov5后处理包括两个主要步骤:类别框的非极大值抑制(NMS)和类别框的偏移量计算。在NMS中,所有的类别框被当做一个类别进行计算,根据设定的IoU阈值进行筛选和合并。具体来说,通过以下步骤执行NMS:
1. 从模型输出中提取类别框的坐标(x,y,w,h)和置信度分数。
2. 计算类别框的偏移量,通过乘以偏移量因子,其中偏移量因子取决于类别的索引和最大宽度高度(max_wh)。这样,偏移量只取决于类别索引,并且足够大。
3. 根据置信度分数对类别框进行排序。
4. 通过使用设定的IoU阈值,应用非极大值抑制,筛选出重叠程度较低的框。
5. 返回经过NMS处理后的类别框。
关于类别框的偏移量计算,具体步骤如下:
1. 从模型输出中提取类别框的坐标(x,y,w,h)和置信度分数。
2. 使用类别索引乘以偏移量因子,其中偏移量因子取决于类别的索引和最大宽度高度(max_wh)。这样,每个类别的偏移量就被计算出来了。
总结起来,Yolov5后处理包括NMS和偏移量计算。NMS将所有的类别框当做一个类别进行计算,根据IoU阈值进行筛选和合并。偏移量计算根据类别的索引和最大宽度高度,计算出每个类别框的偏移量。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [YOLOV5目标检测-后处理NMS(非极大值抑制)](https://blog.csdn.net/qq_22764813/article/details/124401541)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)