YOLO后处理

首发于个人博客 理论分析 system.png YOLO从v2版本开始重新启用anchor box,YOLOv2网络的网络输出为尺寸为[b,125,13,13]的tensor,要将这个Tensor变为最终的输出结果,还需要以下的处理: 解码:从Tensor中解析出所有框的位置信息和类别信息 NMS:筛选最能表现物品的识别框 解码过程 解码之前,需要明确的是每个候选框需要5+class_num个数据
相关文章
相关标签/搜索