Segmentation-based Similarity Search ONNX

Download models

First, download models for segmentation and hashing

cd models
./get_models.sh
cd ..

Build Docker image to run ONNX models

cd docker
git clone https://github.com/microsoft/onnxruntime.git
docker build -t rbss/onnx:1.0 .
cd ..

Run image with jupyter notebook locally (NVIDIA GPU required)

docker run -v $(pwd):/data -p 8888:8888 --runtime=nvidia -it --rm rbss/onnx:1.0

Got to http://localhost:8888/lab

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
assets		assets
configs		configs
docker		docker
models		models
.gitignore		.gitignore
LICENSE		LICENSE
RBSS.ipynb		RBSS.ipynb
README.md		README.md
util.py		util.py