基于轻量化卷积神经网络的输送带实时跑偏检测方法

摘要

本发明实施例提出了一种基于轻量化卷积神经网络的输送带实时跑偏检测方法，包括：生成数据集；通过图像分割网络FCN对图像进行分割，并根据分割的结果建立轻量化卷积神经网络，并生成损失函数；将所述数据集中的图片输入到所述轻量化卷积神经网络，以利用所述轻量化卷积神经网络的卷积层提取出训练集的图片中的输送带的边缘；然后将轻量化卷积神经网络的卷积层提取出图片中的输送带的边缘与验证集中的图片中的输送带的边缘进行验证以进行损失计算，并根据反向传播算法优化模型参数。