SS928的yolov8推理已经完全测试和优化完。目前推理速度是14.8ms。
先说一下onnx模型转换部分。
基本上和hi3519dv500,hi3516dv500是一样的。并没有不同之处,这部分内容在第九篇文章里有详细描述。
链接:https://blog.csdn.net/kwdx2/article/details/135608411
这里有个小注意的地方,新的ultralytics8.2版本最近更新的比较多,导出模型前最好看一下源码,虽然区别不大,改的时候对应自己用的版本改就可以。
另外导出时选择Onnx12的版本,13,14的版本可能会有问题。
导出后的模型结构是这样的。
我训练的模型是两个类的,这里面class_num和自己模型的类别数有关系。
topK,low_score_thresh,num_thresh 这些参数后面都可以在程序里改,不用太在意。