ARM-software / ML-examples

Arm Machine Learning tutorials and examples

Home Page:https://developer.arm.com/technologies/machine-learning-on-arm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

mnist_caffe example : prototxt

Source82 opened this issue · 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?