我正在尝试修改examples/label_image main.cc以将训练过的模型导出到C ++。似乎预期的输入类型是uint8(确切错误:运行模型失败:无效参数:期望arg [0]为uint8但提供了float)。在函数ReadTensorFromImageFile()中,我将转换数据类型更改为uint8,auto float_caster = Cast(root.WithOpName(" uint8_caster"),image_reader,tensorflow :: DataType :: DT_UINT8);但仍然返回的张量是浮点型?这是铸造问题吗?
As,this表示只支持少数数据类型的转换,是不是支持uint8?我在寻找Cast()运算符的正确位置吗?函数调用Cast()实现在哪里?
有关将C ++张量转换为uint8类型的提示吗?