Anchor-based models use predefined anchor boxes to predict the locations and sizes of objects, but the effectiveness of the anchor box is highly dependent on the dataset and domain it is applied. In this paper, we address this problem by automatically setting the anchor boxes optimized for the dataset, employing a genetic algorithm. The proposed method was evaluated on two different datasets with similar distribution in their train and test sets. The approach was applied to various anchor-based models, all of which demonstrated performance enhancements in terms of mean average precision. Specifically, in the benchmark dataset, even a simple anchor-based model performed as well as state-of-the-art model. In addition, since the model learns more efficiently with the chosen anchor boxes, it is proven that our method contributes to the faster convergence of the models. Thus, this study introduces a novel approach for enhancing the performance of object detection models and significantly contributes to the field of anchor box optimization.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Data-Driven Anchor Box Optimization Using Genetic Algorithm

  • Eunbeom Nam,
  • Suyeon Wang,
  • Wooju Kim

摘要

Anchor-based models use predefined anchor boxes to predict the locations and sizes of objects, but the effectiveness of the anchor box is highly dependent on the dataset and domain it is applied. In this paper, we address this problem by automatically setting the anchor boxes optimized for the dataset, employing a genetic algorithm. The proposed method was evaluated on two different datasets with similar distribution in their train and test sets. The approach was applied to various anchor-based models, all of which demonstrated performance enhancements in terms of mean average precision. Specifically, in the benchmark dataset, even a simple anchor-based model performed as well as state-of-the-art model. In addition, since the model learns more efficiently with the chosen anchor boxes, it is proven that our method contributes to the faster convergence of the models. Thus, this study introduces a novel approach for enhancing the performance of object detection models and significantly contributes to the field of anchor box optimization.