勘智K210是嘉楠科技自主研发的一款边缘侧AI芯片,基于RISC-V架构,内置卷积神经网络加速器KPU。KPU可以实现人脸检测、人脸识别、图像识别、图像分类等机器视觉任务,其特性和约束条件如下,有兴趣的开发者可以尝试。
-
网络输入或者中间的feature map尺寸不大于320*256,不小于4×4。
-
与tensorflow支持的数据排布格式NHWC不同,KPU支持的数据排布格式为NCHW。
-
conv2d
Ksize | stride | required tensorflow padding |
---|---|---|
1*1 | 1 | same或valid |
3*3 | 1 | same |
3*3 | 2 | 手动padding一圈,valid |
- depthwise conv2d
Ksize | stride | required tensorflow padding |
---|---|---|
3*3 | 1 | same |
3*3 | 2 | 手动padding一圈,valid |
- 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 |
- 所有通道必须使用相同的激活函数,支持任意类型激活函数。