ML-examples icon indicating copy to clipboard operation
ML-examples copied to clipboard

mnist_caffe example : prototxt

Open Source82 opened this issue 6 years ago • 0 comments

Hi, Please excuse my ignorance. If I was to use the mnist_caffe example for my custom resnet model, at what point will I pass the prototxt file. Obviously the model is passed at line 49 of the code armnn::INetworkPtr network = parser->CreateNetworkFromBinaryFile("model/lenet_iter_9000.caffemodel", { }, // input taken from file if empty { "prob" }); // output node

what i don't understand is where to specify the prototxt file

Will the mnist_caffe example also work for color image - I presume the data part of my prototxt file specifies the channels. Am I right?

Source82 avatar May 17 '19 12:05 Source82