README.md 1.26 KB
Newer Older
Gunho Park's avatar
Gunho Park committed
1
2
# BASNet: Boundary-Aware Salient Object Detection

3
4
5
6
This repository is the unofficial implementation of the following paper. Please
see the paper
[BASNet: Boundary-Aware Salient Object Detection](https://openaccess.thecvf.com/content_CVPR_2019/html/Qin_BASNet_Boundary-Aware_Salient_Object_Detection_CVPR_2019_paper.html)
for more details.
Gunho Park's avatar
Gunho Park committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

## Requirements
[![TensorFlow 2.4](https://img.shields.io/badge/TensorFlow-2.4-FF6F00?logo=tensorflow)](https://github.com/tensorflow/tensorflow/releases/tag/v2.4.0)
[![Python 3.7](https://img.shields.io/badge/Python-3.7-3776AB)](https://www.python.org/downloads/release/python-379/)

## 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