search

rockchip-linux / rknn-toolkit2

BSD 3-Clause "New" or "Revised" License
874 stars 154 forks source link

sdk：1.5.2转换得到的模型会出现检测置信度大于1，1.5.0是正常的 #281

Open fzu-TJT opened 8 months ago

fzu-TJT commented 8 months ago

之前使用1.5.0的rknn-toolkit2转换onnx模型为rknn模型，模型正常；使用1.5.2以及1.6.0版本的转换模型会出现模型检测的置信度大于1，这是为什么？ image

yuyun2000 commented 7 months ago

是不是最后一层没接sigmoid？

fzu-TJT commented 7 months ago

是不是最后一层没有接sigmoid？

好像是的，onnx输出节点前没接sigmoid

yuyun2000 commented 7 months ago

那不就结了，没有sigmoid他怎么输出0-1

fzu-TJT commented 7 months ago

那不就结了，没有sigmoid他怎么输出0-1

这样子啊，那之前版本的为啥可以

yuyun2000 commented 7 months ago

这个你自己再确认一下吧

fzu-TJT commented 7 months ago

这个你自己再确认一下吧

好的，感谢