推理问题 - Githubissues

leafqycc / rknn-cpp-Multithreading

A simple demo of yolov5s running on rk3588/3588s using c++ (about 142 frames). / 一个使用c++在rk3588/3588s上运行的yolov5s简单demo(142帧/s)。

Apache License 2.0

472 stars 87 forks source link

推理问题 #33

Open zyitom opened 10 months ago

zyitom commented 10 months ago

想问下我换上自己的yolov5模型输出却是这样的，框框乱飞是要怎么处理呢

leafqycc commented 10 months ago

想问下我换上自己的yolov5模型输出却是这样的，框框乱飞是要怎么处理呢

您可以使用该网站在线查看模型的结构，确认一下算子是否和demo中的相似（可能存在激活函数不同 silu或relu 的模型）https://netron.app/ 如果相似，那么请去修改include/postprocess.h下第9行的检测种类数量以及src/postprocess.cc下第30-32行的anchor参数（该参数应该在您训练的过程中获取）