2021第九届“泰迪杯”B题岩石分类挑战赛

2021第九届“泰迪杯”B题岩石分类挑战赛

1、总体思路

借助pytorch框架使用轻量卷积神经网络进行分类训练，使用官方4月24日发布的315张图片进行数据增强，对一部分数据进行预处理，投入到网络进行训练得到还算不错的效果，感兴趣的同学可以使用我的网络进行微调，这里只展示测试代码。

个人联系方式见留言板

2、训练结果

模型大小只有16M，准确率可达79.37%，有图有真相✌

3、预测代码

事先说明类别对应的标签如下：

{'深灰色粉砂质泥岩': 0, '浅灰色细砂岩': 1, '深灰色泥岩': 2, '灰色泥质粉砂岩': 3, 
 '灰色细砂岩': 5, '灰黑色泥岩': 4, '黑色煤': 6}

这里选用了 第0类 图像也就是 深灰色粉砂质泥岩 进行预测：