The article describes an approach to the implementation of a software package for determining the probability of a collision of a mobile robot with obstacles. This approach is based on a neural network model with attention. Key feature is the method of the training dataset generation: the labeling of obstacles and the values of the probability of collision with them is performed not in manual mode, but using a deterministic algorithm that uses the result of semantic segmentation using another pre-trained neural network. This method allows to use a poorly detailed description of the external environment for training convolutional neural networks with attention on the example of recognizing obstacles when a mobile robot moves in simulation mode. At the same time, low detail allows to reduce the time-consuming process of manual data labeling due to automatically generated sampling in the NVIDIA Isaac environment, and the attention mechanism allows to increase the interpretability of the analysis results.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.