关于yolov5后处理细节
近日做项目遇到yolov5后处理操作,有点记不太清楚,所以想记录下来,方便后期复习。我们都知道,yolov5模型的输入三个feature map的集合,加上batch的维度,也就是三维张量,即[batch,(p3*p3+p4*p4+p5*p5)*3,4+conf+cls_num],模型输出的为相对于调整图片的xywh,然后后面就要进入后处理阶段。后处理采用了两次阈值筛选,然后送入了nms处理。两次筛选都用的是conf_thres,只有nms求IOU的时候采用的iou_thresh.
原创
2022-04-20 20:55:45 ·
8330 阅读 ·
1 评论