Clear Filters
Clear Filters

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
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.




Community Treasure Hunt

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

Start Hunting!