推理问题
zyitom opened this issue · 1 comments
zyitom commented
想问下我换上自己的yolov5模型输出却是这样的，框框乱飞是要怎么处理呢
leafqycc commented
想问下我换上自己的yolov5模型输出却是这样的，框框乱飞是要怎么处理呢
您可以使用该网站在线查看模型的结构，确认一下算子是否和demo中的相似（可能存在激活函数不同 silu或relu 的模型）https://netron.app/
如果相似，那么请去修改include/postprocess.h下第9行的检测种类数量
以及src/postprocess.cc下第30-32行的anchor参数（该参数应该在您训练的过程中获取）