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

what are the modification i should do to change this problem to solve the multiclass problem(more than 2 class).For Example cat,dog,horse,cow... #2

Open
indrakalita opened this issue Mar 9, 2017 · 7 comments

Comments

@indrakalita
Copy link

No description provided.

@jiexunsee
Copy link
Owner

this other repository i have is probably easier to use and runs faster. Anyway, to do more than 2 classes, you'd need to change the classifier.py script, and do the labelling accordingly for your different classes.

You could also check out the retrain code on the tensorflow github. To use it for multiclass problems you'll have to just put your classes into different folders and run the retrain.py code. The code is rather long though, so it might take a while to understand.

@indrakalita
Copy link
Author

Can you please modify the classifier.py script for multiclass classification

@jiexunsee
Copy link
Owner

If your dataset of images is named like cat001, cat002 and so on then it'll be easy, you could do the same as I did, which was to just attach labels based on the image filenames. Or else you'd have to find some other way to do the labelling, depending on your dataset.

@indrakalita
Copy link
Author

indrakalita commented Mar 10, 2017 via email

@jiexunsee
Copy link
Owner

Apologies I'm a little busy but for now you could look at the create_image_lists function in the tensorflow retrain script. It deals with different classes being stored inrespective different folders. Essentially it uses the gfile module which would probably be what you need.

I'll let you know if I'm able to do it. Also, do use my other repository here as it'll be much faster in producing the bottleneck files.

@indrakalita
Copy link
Author

indrakalita commented Mar 16, 2017 via email

@jiexunsee
Copy link
Owner

you're welcome :)

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

2 participants