You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Serving PyTorch Models in C++ on Windows10 platforms
How to use
Prepare Data
examples/data/train/
- 0
- 1
.
.
.
- n
examples/data/test/
- 0
- 1
.
.
.
- n
Train Model
cd examples && python train.py
Transform Model
cd examples && python transform_model.py
Test Model
cd makefile/pytorch
mkdir build && cd build && cmake -A x64 ..
or
mkdir build && cd build && cmake -G "Visual Studio 15 2017 Win64" ..
set Command Arguments -> ..\..\..\examples\checkpoint ..\..\..\examples\images
set Environment -> path=%path%;../../../thirdparty/libtorch/lib;../../../thirdparty/opencv/build/x64/vc15/bin;
Test CUDA Softmax
cd makefile/cuda
mkdir build && cd build && cmake -A x64 ..
or
mkdir build && cd build && cmake -G "Visual Studio 15 2017 Win64" ..
Inference onnx model
cd makefile/tensorRT/classification
mkdir build && cd build && cmake -A x64 ..
or
mkdir build && cd build && cmake -G "Visual Studio 15 2017 Win64" ..
set Environment -> path=%path%;../../../../thirdparty/TensorRT/lib;
Inference caffe model for faster-rcnn
cd makefile/tensorRT/detection
mkdir build && cd build && cmake -A x64 ..
or
mkdir build && cd build && cmake -G "Visual Studio 15 2017 Win64" ..
set Environment -> path=%path%;../../../../thirdparty/TensorRT/lib;