基于Neon指令的卷积神经网络的算法优化方法及装置

摘要

本发明提供了基于Neon指令的卷积神经网络的算法优化方法，该方法包括：将卷积层的卷积核图像进行矩阵化处理，获取对应的A矩阵，并将A矩阵列数按照4倍数对齐；输入待卷积图像，将待卷积输入图像进行矩阵化处理，获取对应的B矩阵，并将B矩阵行数按照4倍数对齐；对B矩阵进行转置变换，获取转置矩阵Bt；计算A矩阵和Bt矩阵的行与行点积；采用Neon指令进行并行优化处理。与现有技术相比，本发明可以有效地提升卷积神经网络的计算性能。