pytorch转kmodel

Question

1. 将自己训练的pytorch模型转成k210所支持的kmodel；
2. 在之前的尝试中我在.pth -> .onnx -> .pb中，出现转出来的.pb文件是一个文件夹的情况；
3. 虽然可以加载，但是输入图片出来的结果令人匪夷所思，如下图：
   

   

   
   这是原.pth推理的结果：

同时我尝试过各种网上的各种解答，最终没能解决问题，以下是我用的环境：
windows，python=3.7, onnx==1.14.0，onnx-tf==1.14.0, tensorflow==2.4.1, tensorflow-addons==0.12.1
希望能得到解答，更希望能得到.pth转.kmodel的完整解答

Answer 1

参考下同学

K210学习记录（3）——kmodel生成与使用_XS30的博客-CSDN博客 0、引言该博客仅作为学习K210单片机KPU跑目标检测模型记录，本人新人小白，本文旨在备忘，如有错误，还望指出，谢谢。1、模型转换1.1、首先需要下载NNCase工具箱，各版本下载地址1.2、有关nnc的使用，在Windows下首先在运行中，输入CMD，打开命令行窗口1.3、使用cd命令，到nnc的根目录下1.4、后使用nnc相关命令行进行操作，命令行说明可以参考在将yolo.tflite模型放到ncc根目录下后，使用以下命令进行转换ncc yolo.tflite yolo.kmodel -_kmodel https://blog.csdn.net/u014798590/article/details/110549498