2.nncase进行模型转换

我们现在已经将模型从pytorch格式转换为onnx格式，现在我们需要将模型继续转换为nncase格式的模型文件。

所以我们需要使用到nncase中Compiler部分，将ONNX模型转换为DongshanPI-Vision开发板（K510芯片）中使用的kmodel文件。

我们已经提前在Ubuntu中提供好了模型转换程序，程序位于/home/ubuntu/yolov5s-modelTransformation下gen_yolov5s_320_with_sigmoid_bf16_with_preprocess_output_nhwc.py

①将刚刚生成的onnx拷贝到当前目录下，如下图所示：

②执行该程序转换模型：

python gen_yolov5s_320_with_sigmoid_bf16_with_preprocess_output_nhwc.py --target k510  --dump_dir ./tmp --onnx ./yolov5s-sim.onnx --kmodel ./yolov5s-sim.kmodel

阅读nncase开发手册，了解模型转换程序gen_yolov5s_320_with_sigmoid_bf16_with_preprocess_output_nhwc.py

对比提供的API与实际的模型转换程序。打开文档网址：https://canaan-docs.100ask.net/Application/AIApplicationDevelopment-Canaan/05-nncase_Developer_Guides.html

模型转换程序