海思3559 020版本SDK移植问题

Hsdxm / hisi-yolov5

海思设备上部署阉割版yolov5

Apache License 2.0

14 stars 9 forks source link

海思3559 020版本SDK移植问题 #1

Open JimmyLauren opened 3 years ago

JimmyLauren commented 3 years ago

你好，我按照你给的步骤操作了一下，在3559上结果是乱的，请问一下你是在海思的那款芯片和配套SDK上运行的啊？

JimmyLauren commented 3 years ago

你好，还有几个问题想请教一下，我把问题列一下： 1、请问一下pytorch U版yolov5你用到是哪个release版本呀？ 2、转onnx模型的时候需不需要更改opset_version呀？原始的opset_version是12的。 3、你给的last.pt好像是8类别的，不是10类别的。非常期待你的回复。

Hsdxm commented 3 years ago

问题1：下载官方最新代码后，git reset --hard 69be8e738，切换到4.0分支，该分支导出的onnx模型有三个分支，而5.0有四个分支。问题2：切换到4.0分支以后，不需要更改opset_version 问题3：这是一个8类的模型，此处上传模型只是提供一个参考，用于对比focus层的改动。

Hsdxm commented 3 years ago

我在海思3519A和3516D上都测试过，海思提供的代码中存在几处bug，会导致：1）最后一类无法检出；2）输入宽高必须一致。转换成caffemodel后，记得把prototxt文件的每个分支的最后两层删掉。

JimmyLauren commented 3 years ago

非常感谢回复，就是说按照hisi里面示例图片修改海思官方的代码可以解决那两个bug是吧？ PS：在3559上跑你给的例子的时候NMS算法好像没起到作用，同一个位置出现多个同一类别的框，请问有遇到过这个问题嘛？

JimmyLauren commented 3 years ago

未命名1631530101

Hsdxm commented 3 years ago

这个问题我也遇到过，我是在外面重新写了一个NMS

JimmyLauren commented 3 years ago

OK，非常感谢。

JimmyLauren commented 3 years ago

NMS的问题解决了，是因为SVP_NNIE_Yolov2_NonMaxSuppression函数里面有u32Num < u32MaxRoiNum的限制，删掉就OK了