可编程二维图像处理器上的卷积神经网络

摘要

描述了一种方法，其包括在具有执行通道阵列和二维移位寄存器的图像处理器上执行卷积神经网络层。二维移位寄存器为执行通道提供局部相应的寄存器空间。卷积神经网络的执行包括将三维图像数据块的图像数据的平面加载到二维移位寄存器。执行卷积神经网络还包括通过依次进行以下步骤来执行图像数据的平面与系数值阵列的二维卷积：在执行通道内同时相乘相应的像素和系数值以产生部分乘积的阵列；在执行通道内同时将部分乘积和保存在图像数据内不同模版的二维寄存器中的部分乘积的相应累积值求和；以及通过移位二维移位寄存器阵列内的内容来实现执行通道内的二维卷积的值的对齐。