An Edge Computing-Enabled Train Obstacle Detection Method Based on YOLOv3

Abstract

Rail transit is developing towards intelligence which takes lots of computation resource to perform deep learning tasks. Among these tasks, object detection is the most widely used, like track obstacle detection, catenary wear, and defect detection and looseness detection of train wheel bolts. But the limited computation capability of the train onboard equipment prevents running deep and complex detection networks. The limited computation capability of the train onboard equipment prevents conducting complex deep learning tasks. Cloud computing is widely utilized to make up for the insufficient onboard computation capability. However, the traditional cloud computing architecture will bring in uncertain heavy traffic load and cause high transmission delay, which makes it fail to complete real-time computing intensive tasks. As an extension of cloud computing, edge computing (EC) can reduce the pressure of cloud nodes by offloading workloads to edge nodes. In this paper, we propose an edge computing-based method. The onboard equipment on a fast-moving train is responsible for acquiring real-time images and completing a small part of the inference task. Edge computing is used to help execute the object detection algorithm on the trackside and carry most of the computing power. YOLOv3 is selected as the object detection model, since it can balance between the real-time and accurate performance on object detection compared with two-stage models. To save onboard equipment computation resources and realize the edge-train cooperative interface, we propose a model segmentation method based on the existing YOLOv3 model. We implement the cooperative inference scheme in real experiments and find that the proposed EC-based object detection method can accomplish real-time object detection tasks with little onboard computation resources.