请教.wts模型转换成.engine引擎文件出现的报错和警告。

阿硕 · 发表于 2024-6-11 15:06:30

本帖最后由阿硕于 2024-6-11 15:09 编辑

   按照 SpireCV-SDK Wiki 进行了自训练检测模型yolov5s-targetplate.wts的转换，最终生成了engine文件但终端里有一些警告和错误。
   我又尝试用model里提供的已有模型COCO-yolov5s.wts进行转换，发现存在同样的警告和错误，想知道这样算是转换成功了吗？
   我测试了生成的engine文件可以推理，但这些警告和错误是否影响模型推理的速度和精度？（因为测试时感觉帧率偏低，FPS勉强到30）

（两次转换的终端输出内容如图1所示）

amov_daniel · 发表于 2024-6-12 10:14:00

您好如有“build engine successfully!”即表示模型转换成功；

然后，帧率不到30，也可能与你的相机输入帧率有关；比如，相机输入帧率为30，那检测的帧率最大就是30；

其次，以allspark2-orin nx为例，机载的power mode选择对帧率也有一定影响；你可以将其调到“0：MAXN”模式，对阵率也有提升；

最后，关于精度的影响，建议使用我们的精度评估方法具体评估下；
链接如下：https://www.wolai.com/sS3t3zYUNx6gqkQMPfsUg9

阿硕 · 发表于 2024-6-12 17:39:26

amov_daniel 发表于 2024-6-12 10:14
您好如有“build engine successfully!”即表示模型转换成功；

然后，帧率不到30，也可能与你的相机输入 ...

好的，感谢回复。