dkurt / openvino_pytorch_layers

How to export PyTorch models with unsupported layers to ONNX and then to Intel OpenVINO
https://github.com/openvinotoolkit/openvino
Apache License 2.0
26 stars 13 forks source link

Does the grid_sample operation support other arguments? #23

Open DwenGu opened 2 years ago

DwenGu commented 2 years ago

Hi, @dkurt Thanks for your sharing! I wonder whether this op works for other arguments? For example, F.grid_sample(img, grid, mode='bilinear', padding_mode='reflection', align_corners=True).

Will it be supported in iGPU in the future?

Best regards, DwenGu

dkurt commented 2 years ago

Hi, @DunguTmp!

Will take a look at this parameters. For iGPU you might check https://github.com/dkurt/openvino_pytorch_layers/pull/16 by @ngaloppo. I will resolve merge conflict and push to main branch.

DwenGu commented 2 years ago

Hi, @dkurt I have tried the GPU coded provided by @ngaloppo. I use the following command:

benchmark_app.exe -m C:\workspace\LenovoVideoInterp\SupportLayers-Original\grid_sample.xml  -niter 200 -c C:\workspace\LenovoVideoInterp\openvino_pytorch_layers-master\user_ie_extensions\gpu_extensions.xml -d GPU

But, some errors occoured as follows: [Step 1/11] Parsing and validating input arguments [ INFO ] Parsing input parameters [Step 2/11] Loading Inference Engine [ INFO ] GPU extensions is loaded C:\workspace\LenovoVideoInterp\openvino_pytorch_layers-master\user_ie_extensions\gpu_extensions.xml [ INFO ] InferenceEngine: IE version ......... 2021.4.1 Build ........... 2021.4.1-3926-14e67d86634-releases/2021/4 [ INFO ] Device info: [ ERROR ] Failed to create plugin C:\Users\gujun\Documents\Intel\OpenVINO\inference_engine_cpp_samples_build\intel64\Release\clDNNPlugin.dll for device GPU Please, check your environment invalid stoi argument

I don’t know if you have encountered the above problems. Is there a problem with my usage?

Regards, DwenGu.

DwenGu commented 2 years ago

I have already implemented the F.grid_sample(img,` grid, mode='bilinear', padding_mode='reflection', align_corners=True).