| 图片分类 | 训练集数量/张 | 验证集数量/张 | |
| 围岩分级 | II | 426 | 182 |
| III | 426 | 182 | |
| IV | 417 | 178 | |
| V | 444 | 189 | |
| 岩体发育程度 | Joint_1 | 487 | 208 |
| Joint_2 | 532 | 228 | |
| Joint_3 | 542 | 232 | |
| Joint_4 | 511 | 219 | |
| 节理面强度 | Alteration_1 | 437 | 187 |
| Alteration_2 | 446 | 190 | |
| Alteration_3 | 445 | 190 | |
| 赋存条件 | Water_1 | 432 | 185 |
| Water_2 | 427 | 183 | |
| Water_3 | 388 | 166 | |