onnx_test.cpp: main C++ fileCMakeLists.txtflake.nixanddefault.nixfor reproducible buildsbuild.shfor non-nix buildscat_small.jpgoriginal input filecat_sr.pngC++ ONNX outputcat_sr_pytorch.pngPyTorch and Python ONNX output
nix build
# or, if you don't like flakes:
nix-build
You will find the executable in ./result/bin/onnx_test
Dependencies:
- ONNX
- OpenCV
You can use build.sh which just contains:
# create build dir if it does not exist
[ -d build ] || mkdir build
# generate builder
cmake . -B build
# build
cmake --build build<executable_name> <onnx-model> <source-img> <dest-img>
for example:
<executable_name> bsrgan-pretrained.onnx cat_small.jpg cat_sr.png