Компьютерная оптика (Jul 2021)

Investigation of the applicability of the convolutional neural network U-Net to a problem of segmentation of aircraft images

  • D.A. Gavrilov

DOI
https://doi.org/10.18287/2412-6179-CO-804
Journal volume & issue
Vol. 45, no. 4
pp. 575 – 579

Abstract

Read online

The paper investigates the applicability of the convolutional neural network "U-Net" to a problem of segmentation of aircraft images. The neural network image segmentation method is based on the "Carvana" implementation with the "U-Net" architecture. For orientation recognition, a neural network built in the Keras open neural network library based on the pretrained VGG16 neural network is used. The approach considered allows the image segmentation to be conducted. The results of the experiments have shown the possibility of a fairly accurate selection of the object of interest. The resulting binary masks make it possible to visually classify the aircraft in the image.

Keywords