How to use Bumblebee to train a dataset for product recognition in our ecommerce setup?

How would I go about training a dataset for product recognition in our ecommerce setup?

I have some 27 000 nicely categorized pictures

5 Likes

Axon has a training API. So you can get the Bumblebee model and start training it. You will most likely need a GPU. Check some of the notebooks in the Axon repository and some general notes about training for machine learning (i.e. how to split your dataset into train and test and tricks for training images - such as rotating them).

7 Likes