勘智K210 KPU特性及约束

勘智K210是嘉楠科技自主研发的一款边缘侧AI芯片,基于RISC-V架构,内置卷积神经网络加速器KPU。KPU可以实现人脸检测、人脸识别、图像识别、图像分类等机器视觉任务,其特性和约束条件如下,有兴趣的开发者可以尝试。

  1. 网络输入或者中间的feature map尺寸不大于320*256,不小于4×4。

  2. 与tensorflow支持的数据排布格式NHWC不同,KPU支持的数据排布格式为NCHW。

  3. conv2d

Ksize stride required tensorflow padding
1*1 1 same或valid
3*3 1 same
3*3 2 手动padding一圈,valid
  1. depthwise conv2d
Ksize stride required tensorflow padding
3*3 1 same
3*3 2 手动padding一圈,valid
  1. Pooling
Ksize stride required tensorflow padding
2*2 1 same
2*2 2 如果feature map size是ksize的整数倍,same或valid;否则valid
4*4 4 如果feature map size是ksize的整数倍,same或valid;否则valid
  1. 所有通道必须使用相同的激活函数,支持任意类型激活函数。

Published by

风君子

独自遨游何稽首 揭天掀地慰生平