[发明专利]一种基于Winograd算法的3D CNN加速方法及系统

摘要

本发明公开了一种基于Winograd算法的3D CNN加速方法及系统，其方法实施步骤包括：从输入特征图中读取待变换的特征图子块，从权值缓存中读取卷积核子块，对特征图子块Bin以及卷积核子块执行3D Winograd算法输出结果并累加输出累加结果，判断是否输入特征图中的所有输入特征图都读取完毕，如果读取完毕，则将累加结果写回输出特征图缓存Out。本发明通过对Winograd算法进行了扩展并将其用于3D CNN计算中，运用了2D Winograd算法进行CNN加速并取得了良好的效果，能够有效降低CNN算法的计算复杂度、提升基于FPGA的3D CNN加速器的计算性能以及能效比。

主权项

一种基于Winograd算法的3D CNN加速方法，其特征在于实施步骤包括：1)从输入特征图in中读取待变换的特征图子块Bin，从权值缓存w中读取卷积核子块Bw；2)对特征图子块Bin以及卷积核子块Bw执行3D Winograd算法输出结果Tp1；3)对执行3D Winograd算法输出的结果Tp1进行累加，并输出累加结果Sum；4)判断是否输入特征图in中的所有输入特征图都读取完毕，如果尚未读取完毕，则跳转执行步骤1)；否则，跳转执行步骤5)；5)将累加结果Sum写回输出特征图缓存Out。