Accurate segmentation of digital rock images is essential for characterizing pore-matrix systems and predicting petrophysical properties. However, the diversity of rock textures across different lithologies poses a significant challenge for conventional segmentation networks, especially under limited training data. To address this, we introduce DRI-SAM (Digital Rock Image - Segment Anything Model), a hybrid segmentation framework that leverages the powerful visual prior of the Segment Anything Model (SAM) and adapts it to the digital rock domain. Specifically, we apply LoRA-based fine-tuning to SAM's image encoder to better capture rock-specific microstructures, while U-Net is employed to generate prompt points, guiding SAM toward accurate pore-matrix delineation. This approach retains the encoder's representational power while allowing domain-specific adaptation via LoRA, enabling effective cross-domain generalization under limited training data. The model is trained exclusively on 200 annotated images of Bentheimer sandstone, covering two distinct voxel resolutions, and is evaluated on digital rock images of varying lithologies, resolutions and imaging modalities. The results confirm that DRI-SAM achieves accurate segmentation on both sandstone and more challenging carbonate samples, including synthetic and SEM images, without additional retraining or parameter adjustments. Compared to DeepLabV3 + and the only LoRA-tuned SAM, DRI-SAM demonstrates superior performance under limited supervision, highlighting its strong generalization and practical value in digital rock image analysis. Moreover, the findings suggest that foundation models like SAM, when properly adapted, also hold great promise for broader geoscientific imaging tasks.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |