Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

推理问题 #33

Open
zyitom opened this issue Jan 18, 2024 · 1 comment
Open

推理问题 #33

zyitom opened this issue Jan 18, 2024 · 1 comment

Comments

@zyitom
Copy link

zyitom commented Jan 18, 2024

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

20240118_175208.jpg

@leafqycc
Copy link
Owner

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

20240118_175208.jpg

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants