The scarcity of samples seriously restricts the promotion and application of deep learning technology in the field of intelligent recognition of rock and mineral. Therefore, an intelligent incremental method for image samples of rock and mineral based on an improved deep convolutional generative adversarial network (DCGAN) was proposed. By introducing self-attention mechanism to enhance the details and authenticity of generated images, using loss function with gradient penalty mechanism (WGAN-GP) to improve training stability, expanding network structure to enhance model performance, and combining various data augmentation techniques to expand sample diversity and quality, the problem of sample scarcity has been effectively alleviated. The experimental results show that the proposed method outperforms comparative models in evaluation indicators such as peak signal-to-noise ratio, Fréchet initial distance, and structural similarity index. The loss function curves show that the model training process is stable and has a good convergence performance. The clustering analysis results indicate that images generated by the model are superior to the original images in terms of complexity and diversity. The survey results show that the model can generate high-quality and highly realistic images of rock and mineral. The research effectively alleviated the problem of scarcity of rock and mineral image samples, laying the foundation for the intelligent development and practical application of rock and mineral image generation.
| 科 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 |