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

bug in CPP netforward code #14

Open
zhengstake opened this issue Jan 18, 2018 · 6 comments
Open

bug in CPP netforward code #14

zhengstake opened this issue Jan 18, 2018 · 6 comments

Comments

@zhengstake
Copy link
Contributor

out of shear luck, the python inference code I wrote gave me significantly better detection results. So I did a big more debugging on the C++ netforward code under examples/yolo_9000_test.

I found one bug in yolo_v2_output.h in getResult function where the class label scores are compared. The best_score variable is currently declared as int. But it must be float. Otherwise any score less than 1 will treated as 0 and wrong class of object will be selected.

@shabnamghadar
Copy link

@zhengstake would you mind sharing your python inference code?

@zhengstake
Copy link
Contributor Author

You should have seen my reply to aaron0813.

I did a fork to zhengstake/caffe-yolo-9000 and a branch cadence2018. I added a python inference script under caffe-yolo-9000/examples/yolo/voc_mode/yolo_v2_test.py. The script performs inference just as netforward.cpp. But I added visualization step. You can check it out.

I filed a pull request for it to be merged.

@shabnamghadar
Copy link

Thanks

@shabnamghadar
Copy link

@zhengstake I have looked at your prototxt and it seems like if we get rid of route and reorg layer there is nothing out of ordinary there. Can we train the weights in darknet and convert them to caffe?

@zhengstake
Copy link
Contributor Author

I have specific reasons to utilize Caffe flow. So I haven't explored much on the Darknet code base. I suspect if you don't want to make any change to the network, then training with Darknet would be just fine. However, previously I had issues getting the models converted to Caffe for Yolo V1. I'd like to avoid dealing with multiple frameworks.

@karta0807913
Copy link
Owner

thanks I fixed that
and I merge your pull request
I'm sorry I so long to reply

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