Detection and classification of galaxy morphology based on YOLOv5

Feihu Wang; Tao Hu; Minjuan Gu

doi:10.1117/12.2659975

3 February 2023 Detection and classification of galaxy morphology based on YOLOv5

Feihu Wang, Tao Hu, Minjuan Gu

Proceedings Volume 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022); 125111C (2023) https://doi.org/10.1117/12.2659975
Event: Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 2022, Hulun Buir, China

Abstract

Detection and morphological classification of galaxies are important steps to study the formation, structure and evolution of galaxies. However, with the development of large-scale sky surveys such as dark Energy Spectral Survey (DESI), the image data of galaxies are increasing rapidly, and the problem of low classification efficiency caused by excessive image data is difficult to be solved by traditional classification methods. Machine learning algorithms can help astronomers efficiently and automatically detect and classify different galaxies on large astronomical datasets. Therefore, combining with the current popular YOLO visual detection model, this paper proposes a galaxy shape detection model based on YOLOv5.In this paper, affine transformation is firstly carried out on the image, then bilateral filtering and sharpening are used to enhance the data, and for small target detection, the method of reducing the convolution step is adopted. Finally, the YOLOv5 model was used to detect and classify spiral galaxies, elliptical galaxies, lensed galaxies (bar galaxies) and stars in the galaxy images. The mAP@0.5 of the model reached 87.63%, which could accurately detect the positions of different galaxies and classify them.

Citation Download Citation

Feihu Wang, Tao Hu, and Minjuan Gu "Detection and classification of galaxy morphology based on YOLOv5", Proc. SPIE 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 125111C (3 February 2023); https://doi.org/10.1117/12.2659975

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available