Spent 6 hours trying to get an AI model to recognize a coffee mug

So I was playing around with image classification on some open-source model last weekend for a side project. Thought it would take maybe 20 minutes to train it on like 50 photos of my favorite red mug. Six hours later I realized I forgot to resize the images and half of them were 4000x3000 pixels. Has anyone else burned a whole Saturday on something this simple?

2 comments

2 Comments

blake3029d ago

Laughed out loud when I read this cause I did the exact same thing with a plant identifier model... spent a whole Sunday wondering why it kept classifying my succulent as a pizza. Turns out I had the images in a folder labeled "pizza" from an old project and the script was reading the folder names instead of the actual photos. Six hours down the drain for a 30 second fix.

knight.diana8d ago

Feel you hard on that one, especially what @blake302 described - folder names will get you every time.