CPC G06N 3/08 (2013.01) [B61L 23/041 (2013.01); G06F 18/214 (2023.01); G06N 3/04 (2013.01); G06T 7/11 (2017.01); G06V 10/764 (2022.01); G06V 20/58 (2022.01); G06T 2207/10048 (2013.01); G06T 2207/20081 (2013.01); G06T 2207/20084 (2013.01)] | 10 Claims |
1. A method of rails and obstacles detection, the method comprising:
receiving a plurality of image frames from at least one imaging sensor installed on a locomotive of a train, wherein the at least one imaging sensor facing a direction of travel of the train;
determining, for each image frame of a subset of the plurality of image frames, an input data structure comprising the respective image frame and a history element, wherein the history element comprises a segmentation mask prediction determined for an image frame preceding to the respective image frame; and
determining, based on the input data structure of each of the image frames of the subset and using a rails and switches states detection neural network (RSSD NN), a segmentation mask prediction of the respective image frame, wherein the segmentation mask prediction comprises at least a current railway path along which the train travels.
|