This POC is using CNTK 2.1 to train model for multiclass classification of images. Our model is able to recognize specific objects (i.e. toilet, tap, sink, bed, lamp, pillow) connected with picture types we are looking for. It plays a big role in a process which will be used to classify pictures from different hotels and determine whether it's a picture of bathroom, bedroom, hotel front, swimming pool, bar, etc.
MIT License
37
stars
22
forks
source link
Insufficient number of colors in the plot_helper #7
In the plot_helper.py, the color for drawing boxes currently only supports up to 15 classes, which is defined by five base colors and their 3 variants.
But it is not enough for me when i train and evaluate via my own custom dataset, which contains about 50 classes.
In the
plot_helper.py
, the color for drawing boxes currently only supports up to 15 classes, which is defined by five base colors and their 3 variants.But it is not enough for me when i train and evaluate via my own custom dataset, which contains about 50 classes.