Few-Shot intelligent identification of rock thin sections based on SAM
摘要
Rock thin section identification is a complex task, primarily constrained by the extraction of complex minerals and the acquisition of large-scale labeled data. This paper proposes a rock thin section identification method designed for few-shot labeled data, which enables the segmentation and identification of various rock minerals with minimal labeled data. The SAM model is used for mineral particle extraction, combined with the focal loss function, transfer learning, and the integration of multiple classification models to identify thin sections. The prediction process is evaluated at multiple levels. Ultimately, the method achieved the extraction and identification of 11 minerals using only 38 labeled data samples, with an identification accuracy of 91%. This approach significantly reduces the cost of manual labeling, requiring only a small amount of labeled data and minimal training effort to identify specific mineral classes. The source code of the proposed method is available at https://github.com/Xuerenbujianhua/SAMRocks.