image recognition ai

i made an ai that recognizes whether an image is a dog or a cat with some 97% accuracy

it’s not a brand new model, it’s trained based on ResNet 18,but itw orks because

I train it on images of cats and dogs, and then give it a bunch of images to classify, and then repeat the process over and over again

each epoch takes like 5 minutes to complete and i didnt immediately know how to save the model after training so it was a pain retraining and testing the model after every failed attempt because it took like a half an hour to train fully

but i DID eventually figure it out!! and then i made a separate script which loads the saved model and has it look at a couple handpicked images, and decide whether they’re a cat or a dog.

“dog”



it’s very cool, thank you for reading, guten morgen/tag/abend/nacht (im also learning german)

Leave a Reply

Your email address will not be published. Required fields are marked *