all AI news
A Lightweight NMS-free Framework for Real-time Visual Fault Detection System of Freight Trains. (arXiv:2205.12458v1 [cs.CV])
cs.CV updates on arXiv.org arxiv.org
Real-time vision-based system of fault detection (RVBS-FD) for freight trains
is an essential part of ensuring railway transportation safety. Most existing
vision-based methods still have high computational costs based on convolutional
neural networks. The computational cost is mainly reflected in the backbone,
neck, and post-processing, i.e., non-maximum suppression (NMS). In this paper,
we propose a lightweight NMS-free framework to achieve real-time detection and
high accuracy simultaneously. First, we use a lightweight backbone for feature
extraction and design a fault detection …
arxiv cv detection framework free freight real-time time trains