tensorflow学习笔记 inferenceInterface.feed()

举例1:

inferenceInterface.feed(inputName, floatValues, 1, inputSize, inputSize, 3);

第一个参数是输入节点的名称,第二个参数floatValues是输入节点的值,定义的时候是一维float数组;inputSize, inputSize, 3是输入节点的shape,floatValues数组的大小是inputSize*inputSize*3;

举例2:

inferenceInterface.feed(inputName, inputs, WIDTH, HEIGHT);

第一个参数是输入节点的名称,第二个参数inputs是输入节点的值,定义的时候是一维int数组,其大小等于WIDTH*HEIGHT

所以对inputs赋值的时候是从inputs[0]到inputs[WIDTH*HEIGHT-1]

你可能感兴趣的:(tensorflow)