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

训练2类目标 #10

Closed
wait1988 opened this issue Oct 11, 2018 · 3 comments
Closed

训练2类目标 #10

wait1988 opened this issue Oct 11, 2018 · 3 comments

Comments

@wait1988
Copy link

你好,请问用yolov3_lite训练两类目标的时候,网络配置文件该如何修改呢?
我改了conv20,conv21的num_output为18
Yolov3Loss1,Yolov3Loss2的num_class为2
训练报错:
yolov3_layer.cpp:143] Check failed: input_count == tmp_input_count (6498 vs. 7581)
求正确的改法~

@eric612
Copy link
Owner

eric612 commented Oct 11, 2018

@wait1988
num_output 應為 (4+1+2)*3 = 21

@wait1988
Copy link
Author

@eric612 谢谢 傻了我 手误。。。

@18022443868
Copy link

目标探测训练2类,Check failed: input_count == tmp_input_count (27075 vs. 7581),感觉num_output 有好几处,都要进行修改?

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

3 participants