关于depth_wise convolution这个OP

[复制链接] · 发表于 2019-12-7 18:01:49

反应个问题： depth_wise convolution 这个操作RK3399pro是肯定支持的，tf 和 caffe的mobilenet都能跑，但是我基于darknet实现了darknet版本的 depth_wise convolution，训练和预测都挺好用的，但是rk转模型还是出问题，不支持这个OP。其实我基本参照caffe的代码写的，c++ 变成c而已。版主说的那个有自定义OP的文档我一直没找到 (捂脸笑) 。请有相关经验的大神出来指点一下。

只看该作者 · 发表于 2019-12-9 08:48:12

一直在解释类似问题：
1. 硬件OP和软件OP并不是一一对应的
2. 各家框架OP都不是一一对应的，tf和caffe和onnx，写到模型文件里的op并不是一一对应的，甚至有很大区别
3. 以上两点差异，造成转换rknn成功率并不相同。个人使用感觉rknn对caffe和tf的支持比较友好，其他的模型格式只能说是尽可能最大程度去兼容。
3. DW卷积不建议在NPU上用，dw的出现是为了解决conv在嵌入式平台运行慢的问题，牺牲精度换取速度，但我们NPU卷积速度很快，dw conv反而慢，不需要去牺牲这精度搞dw。

只看该作者 · 发表于 2019-12-11 17:08:38

jefferyzhang 发表于 2019-12-9 08:48
一直在解释类似问题：
1. 硬件OP和软件OP并不是一一对应的
2. 各家框架OP都不是一一对应的，tf和caffe和onn ...

谢谢。。。。。。。。。。。