基于多尺度密集连接网络的矿物图像智能识别

Intelligent recognition of mineral images based on multi-scale densely connected convolutional network

  • 摘要: 针对判定矿物种属的传统网络由于计算复杂度高及网络参数量大而导致其应用部署所需环境资源要求高的问题,提出了一种基于多尺度密集连接的网络模型(Multi-Scale Densely connected convolutional Network, MS-DenseNet)用于矿物的智能识别. 首先,为了使网络具有多尺度特征学习能力,在密集连接网络中引入多尺度卷积结构;其次,采用分组卷积策略优化网络结构;最后,在网络尾部采用跳跃连接结构以减少特征信息损失. 在自建矿物数据集上的实验结果显示,网络模型的验证集和测试集准确率分别达到90.54%和88.06%,表明该网络模型具有良好的识别能力.

     

    Abstract: Traditional networks of determining mineral species require high environmental resources for their application deployment due to high computational complexity and large network parameters. To solve this problem, a multi-scale densely connected convolutional network (MS-DenseNet) is proposed. Firstly, in order to make the network have multi-scale feature learning ability, a multi-scale convolution structure is introduced in densely connected networks. Secondly, a group convolution strategy is used to optimize the network structure. Finally, the skip connection structure is used at the end of the network to reduce the loss of feature information. The experimental results on a self-built mineral dataset show that the accuracy of validation set and test set of the network model is 90.54% and 88.06% respectively and the MS-DenseNet has good recognition ability.

     

/

返回文章
返回