[pytorch] Added pytorch subplugin support
authorParichay Kapoor <pk.kapoor@samsung.com>
Wed, 15 May 2019 04:50:57 +0000 (13:50 +0900)
committerMyungJoo Ham <myungjoo.ham@samsung.com>
Thu, 6 Jun 2019 04:59:08 +0000 (00:59 -0400)
commit610b3277604e91bb2bb07f949cd5bf0ba301cd17
tree7f0fbb9b7867b3143490fcc2e8571d4ecbeb1718
parent70eea747b50fecfe519d66a4d9c24de354323cc4
[pytorch] Added pytorch subplugin support

1. Added pytorch subplugin support to run inference of networks
2. Added support to run on gpu for the inference
3. Added test cases for the subplugin

Added a sample file torch saved file used to test the code

V2:
Add pytorch in nnstreamer.mk
install pytorch tests
add nnstreamer-pytorch package

V3:
Disable pytorch for tizen build until its supported

V4:
Disable pytorch tests when disabling pytorch builda

V5:
Corrected variable in spec file
Explanation for enable-pytorch-use-gpu option

V6:
Using mnist based unit-test case instead of imagenet based model
to reduce the model file size in use

V7:
Using input and output properties of tensorfilter than specify in
the model file

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
18 files changed:
debian/control
debian/nnstreamer-pytorch.install [new file with mode: 0644]
ext/nnstreamer/tensor_filter/meson.build
ext/nnstreamer/tensor_filter/tensor_filter_pytorch.c [new file with mode: 0644]
ext/nnstreamer/tensor_filter/tensor_filter_pytorch_core.cc [new file with mode: 0644]
ext/nnstreamer/tensor_filter/tensor_filter_pytorch_core.h [new file with mode: 0644]
gst/nnstreamer/nnstreamer_subplugin.c
jni/nnstreamer.mk
meson.build
meson_options.txt
nnstreamer.ini.in
packaging/nnstreamer.spec
tests/meson.build
tests/nnstreamer_filter_pytorch/checkLabel.py [new file with mode: 0644]
tests/nnstreamer_filter_pytorch/create_mnist_jit.py [new file with mode: 0644]
tests/nnstreamer_filter_pytorch/img/9.png [new file with mode: 0644]
tests/nnstreamer_filter_pytorch/runTest.sh [new file with mode: 0644]
tests/test_models/models/pytorch_lenet5.pt [new file with mode: 0644]