Use Custom OpenVINO Operations抛出错误:what(): Operation: output

Question

我参考官方openvino文档定制了一个类。这是我的班级。从这里复制代码。当我在后处理自定义操作中使用该类时，它会抛出一个错误:what(): Operation: output_name0类型的Identity(op::v0)不受支持

ppp.output().postprocess()
//  .convert_layout("NHWC")
//  .convert_element_type(ov::element::u8)
.custom([](const ov::Output<ov::Node>& node) {
return std::make_shared<TemplateExtension::Identity>(node);
});

在这里我学习如何使用openvino后处理自定义操作。

请帮助我如何使用我的自定义操作符在后处理自定义操作!!

Answer 1

return std::make_shared<TemplateExtension::Identity>(node);不包含任何值。你需要使用return std::make_shared<ov::opset8::Abs>(node);。