FeiGeChuanShu / ncnn_Android_blazeface

This a demo of BlazeFace infer by ncnn
87 stars 26 forks source link

关于模型转换的问题 #9

Open DataXujing opened 2 years ago

DataXujing commented 2 years ago

大佬问一下,mediapipe中的face_landmark_with_attention这个模型你是怎样转换到ncnn的,给个思路,我用了tflite和tfjs两种方式转换出来的和你的不一致。

FeiGeChuanShu commented 2 years ago

https://github.com/PINTO0309/tflite2tensorflow 用这个工具试试

DataXujing commented 2 years ago

按照tflite2tensorflow转成onnx了,但是onnx2ncnn中提示Gather not support,想问一下大佬,是对onnx做了修改?还是生成了saved_model做了修改又转的onnx?

FeiGeChuanShu commented 2 years ago

按照tflite2tensorflow转成onnx了,但是onnx2ncnn中提示Gather not support,想问一下大佬,是对onnx做了修改?还是生成了saved_model做了修改又转的onnx?

我记得好像是有个tf特有的仿射变换的op,导出onnx后会变成一堆琐碎的op,需要把他们删掉,然后分开导出ncnn就行

DataXujing commented 2 years ago

OK,我大体明白您是怎么操作的了,也就是把tf中的这个仿射变换op扔掉然后保存成onnx,然后再导出ncnn,我看到你的代码中仿射变换这部分是放在了一个C++的方法中实现的。

感谢大佬,我已经搞出来了

大佬您也在Qingdao?在那家公司高就?

FeiGeChuanShu commented 2 years ago

在高新区一个小公司

pxEkin commented 1 year ago

OK,我大体明白您是怎么操作的了,也就是把tf中的这个仿射变换op扔掉然后保存成onnx,然后再导出ncnn,我看到你的代码中仿射变换这部分是放在了一个C++的方法中实现的。

感谢大佬,我已经搞出来了

大佬您也在Qingdao?在那家公司高就?

大佬 face_landmark_with_attention的onnx模型可以发一下吗? eduyjj@163.com