Imported Net from Caffe Missing Classification Labels

2 views (last 30 days)
When I import a pre-trained Caffee net with importCaffeNetwork including the .prototxt and .caffemodel files and look at the last layer to see the classification labels it outputs:
ClassificationOutputLayer with properties:
Name: 'ClassificationOutput'
Classes: [class1 class2 class3 class4 class5 class6 class7]
OutputSize: 7
Hyperparameters
LossFunction: 'crossentropyex'
I would assume that these labels are included in the files, so why are they missing here?

Answers (1)

Shivam Sardana
Shivam Sardana on 24 Sep 2019
There is a name-apir input 'Classes' argument which you can pass in importCaffeNetwork. You can refer to this documentation link for more detail:
Hope this help.

Categories

Find more on Statistics and Machine Learning Toolbox in Help Center and File Exchange

Products


Release

R2019a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!