Skip to content
Code for CVPR 2019 paper. BASNet: Boundary-Aware Salient Object Detection
Python
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
__pycache__ basnet May 19, 2019
figures more May 19, 2019
model bn6d_2 Sep 4, 2019
pytorch_iou basnet May 19, 2019
pytorch_ssim
test_data Delete BKN06Z000006_W_big.png Sep 4, 2019
LICENSE Initial commit Mar 29, 2019
README.md
README.md~
basnet_test.py
basnet_train.py
data_loader.py

README.md

BASNet

Code for CVPR 2019 paper 'BASNet: Boundary-Aware Salient Object Detection', Xuebin Qin, Zichen Zhang, Chenyang Huang, Chao Gao, Masood Dehghan and Martin Jagersand.

Contact: xuebin[at]ualberta[dot]ca

Required libraries

Python 3.6
numpy 1.15.2
scikit-image 0.14.0
PIL 5.2.0
PyTorch 0.4.0
torchvision 0.2.1
glob

The SSIM loss is adapted from pytorch-ssim.

Usage

  1. Clone this repo
git clone https://github.com/NathanUA/BASNet.git
  1. Download the pre-trained model basnet.pth and put it into the dirctory 'saved_models/basnet_bsi/'

  2. Cd to the directory 'BASNet', run the training or inference process by command: python basnet_train.py or python basnet_test.py respectively.

We also provide the predicted saliency maps for datasets SOD, ECSSD, DUT-OMRON, PASCAL-S, HKU-IS and DUTS-TE.

Architecture

BASNet architecture

Quantitative Comparison

Quantitative Comparison

Qualitative Comparison

Qualitative Comparison

Citation

@InProceedings{Qin_2019_CVPR,
author = {Qin, Xuebin and Zhang, Zichen and Huang, Chenyang and Gao, Chao and Dehghan, Masood and Jagersand, Martin},
title = {BASNet: Boundary-Aware Salient Object Detection},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2019}
}
You can’t perform that action at this time.