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

How I can make confusion matrix #122

Closed
nagaokatakuya opened this issue Jan 11, 2018 · 4 comments
Closed

How I can make confusion matrix #122

nagaokatakuya opened this issue Jan 11, 2018 · 4 comments

Comments

@nagaokatakuya
Copy link

I finished training my dataset(20 classes) about 4000 images.
Then, to make sure right classification by this model, I tried to make confusion matrix by reference to "predict.py", "frontend.py" and "preprocessing.py".
But in basic-yolo-keras, formats of input datas are different from these in normal keras.
So, it is difficult to make confusion matrix.
I'd like some advice.

@experiencor
Copy link
Owner

Hi @nagaokatakuya! Very sorry for missing out this issue. If you want to evaluate the performance of the object detection model, you can use mAP score instead of confusion matrix (#27)

@nagaokatakuya
Copy link
Author

nagaokatakuya commented Jan 25, 2018

Thanks, but I implemented matrices similar to confusion matrix using annotation file and "predict.py". Refer to center x and y coordinations of boxes of annotation files, I can search for the boxes predicted near to the boxes written in annotation files and count labels predicted about the boxes. The metrices are reasonably good to evaluate the detection model.

mAP score would be helpful to evaluate, then next I will implement it.

@karan2808
Copy link

Hey do you mind sharing your code?

@zmlim
Copy link

zmlim commented May 13, 2020

Hi @karan2808 @nagaokatakuya May I know if you guys had any progress with the confusion matrix code?

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

4 participants