pytorch-cpp

Usage

Download libtorch

$ wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
$ unzip libtorch-shared-with-deps-latest.zip

Tracing model and save model.pt

python to_cpp.py

Build

$ cd pytorch-cpp
$ mkdir build;cd build
$ cmake -DCMAKE_PREFIX_PATH=<path to libtorch> ..
$ make
$ ./example-app ../model.pt <img_path>

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
CMakeLists.txt		CMakeLists.txt
README.md		README.md
dog.png		dog.png
example-app.cpp		example-app.cpp
model.pt		model.pt
output.png		output.png
to_cpp.py		to_cpp.py