This repository contains the codebase for generating a torchscript model for ResNet18 using Pytorch in Python and running inference on the model using libtorch and C++.
- Make sure Docker is installed.
src/model_generation_main.py
: This script generates a torchscript model for ResNet18 inference.src/pytorch_main.cc
: This script loads the torchscript model generated bymodel_generation_main.py
and runs inference on some test data.
- Clone the repository:
git clone https://github.com/KranthiGV/pytorch-inference-pipeline
- Navigate into the project directory:
cd pytorch-inference-pipeline
- Run the docker command:
./run_stack.sh