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 to predict via converted yolo9000 model #35

Open
veya2ztn opened this issue Jan 8, 2019 · 0 comments
Open

How to predict via converted yolo9000 model #35

veya2ztn opened this issue Jan 8, 2019 · 0 comments

Comments

@veya2ztn
Copy link

veya2ztn commented Jan 8, 2019

Follow the instruction, I can already

  1. convert the yolo2 weight (for detecting 80 classes) to keras weight, load the yolo2 model in keras and do correct prediction.
  2. convert the yolo9000 weight (for detecting 9418 classes) to keras weight, load the yolo9000 model in keras but I don't how to predict via this model

For yolo2 model, we know that after run

yolo2_model = load_model(MODEL_PATH.h5)
prediction  = yolo2_model.predict(IMAGE_DATA)

prediction is an (?,?,?,3,num_anchors * (num_classes + 5)) tensor

so if num_anchors=1, num_class=80

we know the prediction's last channel is (box_x,box_y,box_w,box_h,box_confidence, <box_class_probs> )

For yolo2 (80 classes), we know <box_class_probs> is just the probabilities.

But for yolo9000, we know it is an Hierarchical Classificaiton. This is mean the output probabilities here is conditional probabilities.

P(Norfolk)=P(Norfolk|terrier)P(terrier|hunting dog)...P(animal|root)

This is why we need a tree here.

I do know the last value mean in <box_class_probs> of yolo2_model : the absolutly probabilities for each catalogy

I don't know what the last value mean in <box_class_probs> of yolo9000_model, is it conditional probabilities?

Does anyone know the network output meaning for yolo9000?

@veya2ztn veya2ztn changed the title 9k How to predict via converted yolo9000 model Jan 8, 2019
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

1 participant