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 "data=@test.jpg" 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.
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