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.

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

curl -F "[email protected]" http://localhost:8080 -o result.png

• 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.

Requires Python v3.6+

virtualenv venv
venv/bin/activate

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

python main.py

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