tensorflow
/
models
mirror of https://github.com/tensorflow/models.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
models/official/projects/basnet/README.md

1.3 KiB
Raw Permalink Blame History

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

TensorFlow 2.4 Python 3.7

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