YOLOv8_obb的预测流程,主要分预处理模块、推理模块和后处理模块。这里面有很多内容是和目标检测预测流程是重合的,主要区别在于Angle分支、NMS后处理以及regularize_rboxes部分。本文也主要介绍一下这三个模块,其他模块可以结合YOLOv8预测流程-原理解析[目标检测理论篇]一起来看,并且阅读前必须先阅读YOLOv8_obb训练流程-原理解析[旋转目标检测理论篇]-CSDN博客.
1.预处理模块
这一部分主要是对输入图片进行标准化处理的,主要是把输入图片调整成640×640尺寸,但为了计算更快并且避免letterBox后灰边面积过大的影响,只需要把短边调整成32的倍数就行。但是这里的输入图片是(1024,1024)可以直接等比例缩放到(640,640)