haruishi43 / boundary_detection

Semantic Boundary Detection

boundary_detection

Based on https://github.com/Lavender105/DFF

Setup

Preprocess Cityscapes

python preprocess/preproc_original.py

Training

CUDA_VISIBLE_DEVICES=0,1 python scripts/train.py --checkname dff_test --base-size 640 --crop-size 640 --worker 4 --batch-size 2

Test

CUDA_VISIBLE_DEVICES=0,1 python scripts/test.py --dataset cityscapes --model dff --checkname trained_dff --resume-dir results/model_best.pth.tar --workers 4 --backbone resnet50 --eval

For visualization:

CUDA_VISIBLE_DEVICES=0,1 python scripts/test.py --dataset cityscapes --model dff --checkname trained_dff --resume-dir results/model_best.pth.tar --workers 4 --backbone resnet50

Evaluate

Currently, MATLAB is necessary to run the evaluation code.