MPolaris / onnx2tflite

Tool for onnx->keras or onnx->tflite. Hope this tool can help you.
Apache License 2.0
526 stars 42 forks source link

用了博主说的官方onnx-tf和这个onnx2tflite做了yolox_tiny的对比 #72

Closed zxccsssd closed 4 months ago

zxccsssd commented 4 months ago

yolox_voc_tiny tflite yolox tflite yolox_voc_tiny onnx

第一幅图是用的是本项目的,第二个是用的官方的,可以看出来确实少了很多transpose操作。这方面我属于小白,我想请教下,本项目中的结尾处也有很多transpose,因为我又查看了图三的onnx模型图,后边是没有那么多transpose,所以说本项目中的结尾处是不是也有多余的?还有就是导出来的文件大小,本项目的文件大小是19M,官方的是5M,图里没有显示量化。文件的大小有什么说法吗?

MPolaris commented 4 months ago

从这个case看,结尾处的tranpose确实有多余。主要是需要对齐通道,现在强制在reshape的时候使用transpose转换回channel-first来进行,你也可以针对特定的情况(比如这里)写一个对应的判断从而不发生,欢迎PR。