arXiv reaDer
AttnConv-Netに基づく自動レールコンポーネント検出
Automatic Rail Component Detection Based on AttnConv-Net
鉄道画像を使用した主要な鉄道コンポーネントの自動検出は、鉄道輸送の安全性を確保するために役立ちます。この論文では、レール、クリップ、ボルトなどの複数のレールコンポーネントを検出するために、注意力を利用した深い畳み込みネットワーク(AttnConv-net)を提案します。提案された方法は、バックボーンとしての深い畳み込みニューラルネットワーク(DCNN)、カスケードアテンションブロック(CAB)、および2つのフィードフォワードネットワーク(FFN)で構成されます。バックボーンから抽出された潜在的特徴の情報を充実させるために、2種類の位置埋め込みが適用されます。処理された潜在的特徴に基づいて、CABは、カテゴリやコンポーネントの境界など、レールコンポーネントのローカルコンテキストを学習することを目的としています。最終的なカテゴリと境界ボックスは、並行して実装された2つのFFNを介して生成されます。小さなコンポーネントの検出を強化するために、トレーニングプロセスではさまざまなデータ拡張方法が採用されています。提案されたAttnConv-netの有効性は、1つの実際のデータセットと別の合成されたデータセットで検証されます。従来の畳み込みニューラルネットワークベースの方法と比較して、提案された方法は、前処理と後処理の必要性を排除することによって検出パイプラインを簡素化します。これにより、より高速でより正確な画像ベースのレールコンポーネント検出を可能にする新しい速度品質ソリューションが提供されます。
The automatic detection of major rail components using railway images is beneficial to ensure the rail transport safety. In this paper, we propose an attention-powered deep convolutional network (AttnConv-net) to detect multiple rail components including the rail, clips, and bolts. The proposed method consists of a deep convolutional neural network (DCNN) as the backbone, cascading attention blocks (CAB), and two feed forward networks (FFN). Two types of positional embedding are applied to enrich information in latent features extracted from the backbone. Based on processed latent features, the CAB aims to learn the local context of rail components including their categories and component boundaries. Final categories and bounding boxes are generated via two FFN implemented in parallel. To enhance the detection of small components, various data augmentation methods are employed in the training process. The effectiveness of the proposed AttnConv-net is validated with one real dataset and another synthesized dataset. Compared with classic convolutional neural network based methods, our proposed method simplifies the detection pipeline by eliminating the need of prior- and post-processing, which offers a new speed-quality solution to enable faster and more accurate image-based rail component detections
updated: Thu Feb 10 2022 07:33:30 GMT+0000 (UTC)
published: Thu Aug 05 2021 07:38:04 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アソシエイト