karolzak / cntk-hotel-pictures-classificator

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

Name mismatching for the base model VGG16 #4

Open TreeLLi opened 6 years ago

TreeLLi commented 6 years ago

Hello,

There is an error of name mismatching when I wanted to switch the base model to the VGG16. Basically, the problem is that you define the name of model by yourself in the config file, e.g. VGG16, while the real name of VGG model referred by the download is VGG16_ImageNet_Caffe. Then after finish downloading, the program named the model as VGG16 but the other part of main program loads the model via the name VGG16_ImageNet_Caffe.

Obviously, the problem will also happen when loading and using other base models if the program-defined name is different from the name defined by others.

Best wishes! Lin.