Skip to content
The code for the CVPR2019 paper Bi-Directional Cascade Network for Perceptual Edge Detection
Branch: master
Clone or download
pkuCactus Update README.md
provided the link of model to google drive.
Latest commit 714848e Mar 10, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
caffemodel2pytorch First version of release code Mar 4, 2019
datasets First version of release code Mar 4, 2019
.gitignore
LICENSE Create LICENSE Mar 5, 2019
README.md
ablation.py First version of release code Mar 4, 2019
ablation_test.py First version of release code Mar 4, 2019
ablation_train.py
ablation_vgg16_c.py
bdcn.py First version of release code Mar 4, 2019
cfg.py
log.py First version of release code Mar 4, 2019
test.py
test_ms.py
train.py
vgg16_c.py

README.md

Bi-Directional Cascade Network for Perceptual Edge Detection(BDCN)

This paper proposes a Bi-Directional Cascade Network for edge detection. By introducing a bi-directional cascade structure to enforce each layer to focus on a specific scale, BDCN trains each network layer with a layer-specific supervision. To enrich the multi-scale representations learned with a shallow network, we further introduce a Scale Enhancement Module (SEM). Here are the code for this paper.

Prerequisites

  • pytorch >= 0.2.0(Our code is based on the 0.2.0)
  • numpy >= 1.11.0
  • pillow >= 3.3.0

Train and Evaluation

  1. Clone this repository to local
git clone https://github.com/pytorch/pytorch.git
  1. Download the imagenet pretrained vgg16 pytorch model [vgg16.pth](link: https://pan.baidu.com/s/10Tgjs7FiAYWjVyVgvEM0mA code: ab4g) or the caffemodel from the model zoo and then transfer to pytorch version. You also can download our pretrained model for only evaluation. The google drive link.

  2. Download the dataset to the local folder

  3. running the training code train.py or test code test.py

Pretrained models

BDCN model for BSDS500 dataset and NYUDv2 datset of RGB and depth are availavble on Baidu Disk.

The link https://pan.baidu.com/s/18PcPQTASHKD1-fb1JTzIaQ
code: j3de

The pretrained model will be updated soon.

You can’t perform that action at this time.