Skip to content

ONNX model for Segmentation-based Similarity Search

License

Notifications You must be signed in to change notification settings

umr-ds/rbss-onnx

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Segmentation-based Similarity Search ONNX

Download models

First, download models for segmentation and hashing

cd models
./get_models.sh
cd ..

Build

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

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

About

ONNX model for Segmentation-based Similarity Search

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published