首页 > 解决方案 > Tensorflow:如何将对象识别模型输入类型设置为浮点数?

问题描述

我有一个只接受uint8作为输入类型的对象识别模型:

--input_layer type=uint8

我想将其更改为接受float作为输入类型。有没有办法为此目的修改当前模型,或者在我的训练中进行一些更改,以便我未来的模型将float作为输入类型?

谢谢

标签: pythontensorflow

解决方案


有这种方法tf.cast(...),但我不知道如何使用它来修改我的模型。如果有办法使用它来解决我的问题,我们将不胜感激。

主持人注意:“记录您自己为回答您的问题所做的持续努力。” 这是关于如何让我的问题更容易回答的建议,所以请不要删除这个


推荐阅读