A project to allow sellers to upload images of their product, and have the trained neural network tag and classify the product
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
Client
.gitignore
README.md
checkpoint.pth.tar
model_best.pth.tar
network.py
nohup_backup.out

README.md

Deep Learning Product Image Tagger

-To start training: python network.py --epochs [NUM_EPOCHS] --pretrained [DIR_TO_IMAGES]

-To start backend server: python network.py --server data

/Client - Front End

-To start frontend server: node server

network.py - Training script