YOLO-Crater Model for Small Crater Detection

Craters are the most prominent geomorphological features on the surface of celestial bodies, which plays a crucial role in studying the formation and evolution of celestial bodies as well as in landing and planning for surface exploration. Currently, the main automatic crater detection models and datasets focus on the detection of large and medium craters. In this paper, we created 23 small lunar crater datasets for model training based on the Chang'E-2 (CE-2) DOM, DEM, Slope, and integrated data with 7 kinds of visualization stretching methods. Then, we proposed the YOLO-Crater model for Lunar and Martian small crater detection by replacing EioU and VariFocal loss to solve the crater sample imbalance problem and introducing a CBAM attention mechanism to mitigate interference from the complex extraterrestrial environment. The results show that the accuracy (P = 87.86%, R = 66.04%, and F1 = 75.41%) of the Lunar YOLO-Crater model based on the DOM-MMS (Maximum-Minimum Stretching) dataset is the highest and better than that of the YOLOX model. The Martian YOLO-Crater, trained by the Martian dataset from the 2022 GeoAI Martian Challenge, achieves good performance with P = 88.37%, R = 69.25%, and F1 = 77.65%. It indicates that the YOLO-Crater model has strong transferability and generalization capability, which can be applied to detect small craters on the Moon and other celestial bodies.