deepb1t/basnet-http

HTTP service wrapper for BASNet: Boundary-Aware Salient Object Detection

BASNet HTTP

This is an HTTP service wrapper for BASNet: Boundary-Aware Salient Object Detection code

The deploy folder contains configuration files for deployment as serverless container with Knative.

Usage:

docker run --rm -p 8080:80 gcr.io/cyrildiagne-ml/basnet-http

Test:

curl -F "data=@test.jpg" http://localhost:8080 -o result.png

Development

• Clone this repository: git clone https://github.com/cyrildiagne/BASNet-http.git
• Go into the cloned directory: cd BASNet-http
• Clone the BASNet repository
• Download the pretrained model basnet.pth
• Put the file inside the BASNet/saved_models/basnet_bsi/ folder.

Build from source:

Option 1 - Locally with virtualenv

virtualenv venv
venv/bin/activate

pip install torch==0.4.1
pip install -r requirements.txt

python main.py

Option 2 - Using Docker

After you've retrieved the BASNet model.

Download Resnet checkpoint

curl https://download.pytorch.org/models/resnet34-333f7ec4.pth -o resnet34-333f7ec4.pth

docker build -t basnet .
docker run --rm -p 8080:8080 basnet