Skip to content

SebastiaanJohn/sneaker_classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Sneaker Classifier CNN

See the CNN in action: sneakername

A convolutional neural network that classifies up to 900 sneakers with 90% accuracy. Trained using the fast.ai libary.

How this was build

First I scraped the StockX website to get a large csv file of product data. I modifed an existing script from zpencerguy to create a new script that worked on the StockX website. See here: StockX-scraper.

Next, I cleaned the data and put all the product names in a text file, and wrote an Python program to download 100 images per sneaker from Google. This got me 90k images total. Then, I cleaned the data manually which left me with 52k usable images.

Finally, I used the Fastai libary to train a convolutional neural network on the images and got an decent accuracy of 90%.

It's far from perfect, but it was a fun experiment.

About

Convolutional neural network sneaker classifier.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published