vonclites / squeezenet

Tensorflow implementation of SqueezeNet.
MIT License
129 stars 63 forks source link

How can I retrain it with new classes and labels? #9

Open WuXinyang2012 opened 6 years ago

WuXinyang2012 commented 6 years ago

Hi, Thanks for your excellent work! I tried to implement it on a raspberry pi for a kitchen system to detect fruits and vegetables. Thus I need to retrain the network for around 20 classes. I already prepared the dataset from ImageNet but I am not sure how can I feed them to your scripts?