A New Remote Sensing Image Retrieval Method Based on CNN and YOLO

A New Remote Sensing Image Retrieval Method Based on CNN and YOLO

<>Retrieving remote sensing images plays a key role in RS fields, which activates researchers to design a highly effective extraction method of image high-level features. However, despite the advanced features of the image can be extracted by deep-learning method, features fail to cover the overall information due to its rich and complex background. Extract key regions from RS images, recede background interference and retrieval accuracy needs to be solved. Combined YOLOv5 target recognition algorithm with deep-learning method, this paper proposes a novel retrieval method based on target critical region detection. Firstly, the key retrieval regions have been identified. YOLOv5 target recognition algorithm has been used to identify the key regions of the image and served as the retrieval regions. Secondly, the retrieval characteristics are determined. Combining with the CNN model ResNet50, the retrieval features are extracted from the retrieval regions acquired in the previous step, in addition, PCA method has been used to reduce the dimension of the retrieval features. Finally, using weighted distance based on class probability to measure the similarity between a query and retrieve images. Experimental results show that the proposed method can extract better image retrieval features and improve the retrieval performance of RS image.<>