Remote sensing image ship detection method and device based on attention model

Abstract

The invention discloses a remote sensing image ship detection method and device based on an attention model, and the method comprises the steps: 1, collecting a ship remote sensing image, and carrying out the data expansion of an image data set through annotation information; step 2, preprocessing the collected remote sensing image to obtain a data set used for ship detection model training; 3, inputting the images in the training data set into a YOLOV5 attention model improved for a small-size target, and training to obtain a trained remote sensing ship detection model; 4, cutting a remote sensing image to be detected, inputting the cut remote sensing image into the trained remote sensing ship detection model, and outputting a bounding box and confidence of a ship; and mapping bounding boxes of all the cut images back to the original remote sensing image, and obtaining a final detection result after filtering repeated targets through confidence threshold filtering and non-maximum suppression. According to the invention, the problem of difficult detection caused by sparse ship distribution and too small size in the remote sensing image is solved.

Description

technical field [0001] The invention belongs to the fields of computer vision and target detection, and in particular relates to a remote sensing image ship detection method and device based on an attention model. Background technique [0002] In recent years, with the continuous development of intelligent science and technology, and the increasing strategic status of the ocean, in order to improve the ability to supervise and control the ocean, countries have gradually increased the research on ocean monitoring. As an important tool in marine transportation, the positioning of ships in key sea areas is of great significance in the fields of national defense and trade. Since most of the scenes detected and identified by ships are in the vast sea, images are usually collected by means of remote sensing satellites or drones. However, in remote sensing or aerial photography scenarios, the collected images usually have problems such as too small ship targets, sparse distributio...

AI Technical Summary

Problems solved by technology

However, in remote sensing images, ship targets are mostly extremely small targets, and it is difficult to effectively extract rich detailed information. Therefore, it is necessary to modify the target detection network to make it more focused on the recognition of small targets.

Images