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

Using darknet19 as a simple classifier #183

Closed
karthiksprakash opened this issue Mar 26, 2020 · 0 comments
Closed

Using darknet19 as a simple classifier #183

karthiksprakash opened this issue Mar 26, 2020 · 0 comments

Comments

@karthiksprakash
Copy link

I have some newbie questions. I used the darknet19 model to load the corresponding pre-trained weights from the Darknet website (ImageNet 1000 classes). I see the last softmax layer of this model is of shape (None, 8, 8, 1000).

  1. How do I use this for simple classification? The prediction of one of the sample images (eagle.jpg) is of shape - (1, 8, 8, 1000). How do I interpret this? Do I look for the max value of the 1000 elements across the 8x8 3D matrix?

  2. Could this be flattened into 1x1000 array? Would that require re-training?

  3. I want to retrain for a new dataset of 200 classes. Change I change the last softmax layer to shape 200 and train by freezing the other layers?

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