Swift CoreML自动将int转换为float

我是coremltool的新用户，目前我能够转换我的TF模型，并在python中运行它。但是，我的模型的输出层是浮动的，而不是int32。我必须在python中执行回滚转换

model.predict({"Input": image}).astype(int)

有人知道如何取消自动转换吗？并将我的输出保持为多维数组-int

谢谢,

---

++t

您可以使用coremltools Python库将模型输出的数据类型更改为INT32。这将自动将这些数字从浮点数转换为整数

大概是这样的：

import coremltools
model = coremltools.models.MLModel("YourModel.mlmodel")
spec = model._spec
spec.description.output[0].type.multiArrayType.dataType = coremltools.proto.FeatureTypes_pb2.ArrayFeatureType.INT32
coremltools.models.utils.save_spec(spec, "YourNewModel.mlmodel")

---

之所以不自动执行此操作，是因为通常浮点数是正确的数据类型。

谢谢，而且您的书很好。我上周买的。