2

我正在尝试将 Detectron 2 模型转换为 Core ML。我已经编写了脚本,现在我有一个“torch.jit._script.RecursiveScriptModule”类型的模型。如果我尝试使用以下代码将其转换为 Core ML:

mlmodel = ct.converters.convert(
  torchscipt_model,
  inputs=[ct.TensorType(shape=(1, 3, 64, 64))],
)

我收到以下错误:

RuntimeError:在降低图形时遇到未知类型的布尔值。ONNX 导出不支持此类型。

我使用以下代码编写了模型脚本(如 Detectron 的文档中所示)

model = build_model(cfg)
model.eval()

fields = {"pred_boxes": Boxes, "scores": torch.Tensor, "pred_classes": torch.Tensor, "pred_masks": torch.Tensor, "proposal_boxes": Boxes, "objectness_logits": torch.Tensor}

torchscipt_model = scripting_with_instances(model, fields)

您知道将我的 Detectron2 模型转换为 Core ML 的其他方法吗?

非常感谢你的帮助!

4

0 回答 0