Skip to content

Cats & Dogs — Training & Classifying w/ Deep Learning

Pre-release
Pre-release
Compare
Choose a tag to compare
@MarxSoul55 MarxSoul55 released this 26 Mar 04:29
· 414 commits to master since this 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:

  1. 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 through model.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.
  2. You can now classify an image from a URL. Ensure that the URL points to the image by itself!
  3. 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! ❤️