Cats & Dogs — Training & Classifying w/ Deep Learning
Pre-release
Pre-release
Hello World! 😃
In this release, I have refactored the code to make it much more beautiful (in my opinion), and I've added three new features:
- You can now classify all the images in a directory (excluding subdirectories). You can access this through the CLI, but the output is quite ugly—it's simply a dictionary printed out with the
print
function. The reason for this ugliness is that I've only intended the CLI to be for testing purposes, so I don't bother optimizing its looks. If you want the actual dictionary (which is of format{filename: prediction}
), you access it throughmodel.classify
. And of course, if you like the CLI, you're more than welcome to change the way the dictionary prints out yourself, which would be a very trivial task. - You can now classify an image from a URL. Ensure that the URL points to the image by itself!
- I added a safety prompt that makes you confirm that you want to start training (or resume training) on a model. I added it because training replaces the parameters of the old model; if one were to mistakenly forget this, the prompt is there to protect them!
I have put directions for both new features in the new README.md
.
Happy classifying! ❤️