Intel® Distribution of OpenVINO™ Toolkit
Community assistance about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all aspects of computer vision-related on Intel® platforms.
6404 Discussions

A tutorial on how to Create Extensions and Run custom layers on it

Malhar
Employee
481 Views

Hi all,

I am looking for a tutorial / code snippets to create a custom extension and custom layer for Inference Engine. I want to run custom layer on the device other than CPU. I tried following hello_shape_infer_ssd sample code, but couldn't make it through. Is there any other implementation for the same ? (any CPP or Python code implementation)

0 Kudos
2 Replies
Shubha_R_Intel
Employee
481 Views

Please study deployment_tools\model_optimizer\extensions\front\caffe\argmax_ext.py

and also deployment_tools\model_optimizer\extensions\ops\argmax.py

and also deployment_tools\inference_engine\src\extension\ext_argmax.cpp

Also look at http://caffe.berkeleyvision.org/tutorial/layers/argmax.html code. 

 

0 Kudos
Fritz_B_Intel
Employee
481 Views

Additional tutorial for custom layers can be found here: https://github.com/david-drew/OpenVINO-Custom-Layers

 

0 Kudos
Reply