基于动态样本平衡算法的无锚框目标检测模型及应用

A SAMPLE-BALANCED ANCHOR-FREE OBJECT DETECTOR AND ITS APPLICATION

  • 摘要: 针对目标检测模型正负样本数量不均衡的问题,提出一种动态样本平衡算法,并构建无锚框的目标检测模型。模型基于特征图中特征点与目标框的位置,筛选出负责预测目标框的特征点,并计算其代价矩阵,再为图像中的目标动态分配正负样本。因此,模型无需手工设置锚框,还提高了可检出目标的数量和检测精度。在VOC数据集中,模型可以获得91.3%的平均精度,检测速度达到32.1帧/s。将模型应用于机场机坪保障作业中,可以取得89.37%的平均精度,对关键目标的检测精度均在90%以上。

     

    Abstract: To solve the problem of object detector that negative samples are much more than positive samples, a dynamic sample balancing algorithm is proposed. And based on this algorithm, an anchor free object detection model was constructed. Based on the position of feature points and object boxes in the feature map, the model selected the feature points responsible for predicting the object, calculated their cost matrix, and dynamically assigned positive and negative samples to the objects in the image. Therefore, the model did not require manual setting of anchor boxes, and it also improved the number of detectable targets and detection accuracy. In the VOC dataset, the model achieved an average accuracy of 91.3% and a detection speed of 32.1 FPS. Applying the model to airport apron support operations could achieve an average accuracy of 89.37%, and the detection accuracy of key targets was above 90%.

     

/

返回文章
返回