使用自定義 OpenVINO 操作拋出錯誤：what(): Operation: output_name0 of type Identity(op::v0) is not supported

Question

我參考openvino官方文檔定制了一個class。 這是我的 class。 代碼從這里復制。 當我在后處理自定義操作中使用這個 class 時，它會拋出一個錯誤：what(): Operation: output_name0 of type Identity(op::v0) is not supported

    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 Post-processing Customize操作。

請幫助我如何在后處理自定義操作中使用我的自定義運算符！！

Answer 1

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