BASNet: Boundary-Aware Salient Object Detection

This repository is the unofficial implementation of the following paper. Please see the paper BASNet: Boundary-Aware Salient Object Detection for more details.

Requirements

Train

shell

$ python3 train.py \
  --experiment=basnet_duts \
  --mode=train \
  --model_dir=$MODEL_DIR \
  --config_file=./configs/experiments/basnet_dut_gpu.yaml

Test

shell

$ python3 train.py \
  --experiment=basnet_duts \
  --mode=eval \
  --model_dir=$MODEL_DIR \
  --config_file=./configs/experiments/basnet_dut_gpu.yaml
  --params_override='runtime.num_gpus=1, runtime.distribution_strategy=one_device, task.model.input_size=[256, 256, 3]'

Results

Dataset	maxF<sub>β</sub>	relaxF<sub>β</sub>	MAE
DUTS-TE	0.865	0.793	0.046