Intel® Distribution of OpenVINO™ Toolkit
Community support and discussions about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all things computer vision-related on Intel® platforms.
5962 Discussions

Converting a TensorFlow model (Layer not supported)

Di_Fiore__Cristian
186 Views

Hi, I'm trying to convert a TensorFlow model but when I have to freeze the model I can't choose my output node correctly because my layer isn't supported. How can I replace "tf.image.encode_png" with the Model Optimizer? I'd like to know if it's possible to freeze the model before the "encoding" so in my application the result of inference(after the infer request with OV) becomes the input for the last layer in TensorFlow(tf.image.encode_png("my output")).

Thanks.

0 Kudos
1 Reply
SuryaPSC_Intel
Employee
186 Views

Hi Cristian,

You can try to implement the layer tf.image.encode_png as custom layer.

Or you may try dropping the last layer of the model and generate IR from rest of the model, then feed the output of Inference to the layer tf.image.encode_png.

Best Regards,

Surya

Reply