Tensorflow/models/official/projects/basnet/README.md

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

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

Test

$ 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β	relaxFβ	MAE
DUTS-TE	0.865	0.793	0.046