基于YOLO框架的图像目标检测方法

摘要

本发明公开了一种基于YOLO框架的图像目标检测方法。该方法通过建立YOLO框架模型，将待测图像转换成固定尺寸大小的图像后输入YOLO框架模型得到多个预测张量值，每个预测张量值通过转换公式反向计算得到预测矩形框，将得到的多个预测矩形框通过非最大抑制算法处理，得到最可靠的矩形框，并将其转换到原图上即可得到目标的类别和位置，YOLO框架模型是在原有的YOLO框架基础上，对其进行改进，使得检测能力大大提升，使得原有检测不到的小目标能够检测到，并且能够准确识别目标是什么和目标的大致位置，同时相比较于传统的建模方法，减少了开发时间，大大提高了检测的准确度和速度，使得算法能够达到实时的能力。适合在数据处理技术领域推广应用。