arXiv reaDer
FCOSR:空中物体検出用のシンプルなアンカーフリー回転検出器
FCOSR: A Simple Anchor-free Rotated Detector for Aerial Object Detection
既存のアンカーベース指向のオブジェクト検出方法は驚くべき結果を達成しましたが、これらの方法にはいくつかの手動プリセットボックスが必要であり、追加のハイパーパラメータと計算が導入されます。既存のアンカーフリーメソッドは通常、複雑なアーキテクチャを備えており、展開は簡単ではありません。私たちの目標は、航空画像検出のためのシンプルで展開が容易なアルゴリズムを提案することです。この論文では、FCOSに基づく1ステージのアンカーフリー回転物体検出器(FCOSR)を紹介します。これは、ほとんどのプラットフォームに展開できます。 FCOSRは、畳み込み層のみで構成される単純なアーキテクチャを備えています。私たちの仕事は、トレーニングフェーズのラベル割り当て戦略に焦点を当てています。楕円中心サンプリング法を使用して、方向付けされたバウンディングボックス(OBB)に適したサンプリング領域を定義します。ファジーサンプル割り当て戦略は、重複するオブジェクトに適切なラベルを提供します。不十分なサンプリングの問題を解決するために、マルチレベルのサンプリングモジュールが設計されています。これらの戦略は、トレーニングサンプルにより適切なラベルを割り当てます。私たちのアルゴリズムは、DOTA1.0、DOTA1.5、およびHRSC2016データセットでそれぞれ79.25、75.41、および90.15mAPを達成します。 FCOSRは、シングルスケール評価で他の方法よりも優れたパフォーマンスを示します。軽量のFCOSRモデルをTensorRT形式に変換します。これにより、DOTA1.0で73.93 mAPが達成され、Jetson XavierNXでシングルスケールの速度が10.68FPSになります。コードはhttps://github.com/lzh420202/FCOSRで入手できます。
Existing anchor-base oriented object detection methods have achieved amazing results, but these methods require some manual preset boxes, which introduces additional hyperparameters and calculations. The existing anchor-free methods usually have complex architectures and are not easy to deploy. Our goal is to propose an algorithm which is simple and easy-to-deploy for aerial image detection. In this paper, we present a one-stage anchor-free rotated object detector (FCOSR) based on FCOS, which can be deployed on most platforms. The FCOSR has a simple architecture consisting of only convolution layers. Our work focuses on the label assignment strategy for the training phase. We use ellipse center sampling method to define a suitable sampling region for oriented bounding box (OBB). The fuzzy sample assignment strategy provides reasonable labels for overlapping objects. To solve the insufficient sampling problem, a multi-level sampling module is designed. These strategies allocate more appropriate labels to training samples. Our algorithm achieves 79.25, 75.41, and 90.15 mAP on DOTA1.0, DOTA1.5, and HRSC2016 datasets, respectively. FCOSR demonstrates superior performance to other methods in single-scale evaluation. We convert a lightweight FCOSR model to TensorRT format, which achieves 73.93 mAP on DOTA1.0 at a speed of 10.68 FPS on Jetson Xavier NX with single scale. The code is available at: https://github.com/lzh420202/FCOSR
updated: Sun Nov 21 2021 09:49:13 GMT+0000 (UTC)
published: Sun Nov 21 2021 09:49:13 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト