/BDCN

Reproducing the code for the CVPR2019 paper Bi-Directional Cascade Network for Perceptual Edge Detection

Primary LanguagePythonMIT LicenseMIT

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/pkuCactus/BDCN.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