HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Pixel Difference Networks for Efficient Edge Detection

Zhuo Su Wenzhe Liu Zitong Yu Dewen Hu Qing Liao Qi Tian Matti Pietikäinen Li Liu

Pixel Difference Networks for Efficient Edge Detection

Abstract

Recently, deep Convolutional Neural Networks (CNNs) can achieve human-level performance in edge detection with the rich and abstract edge representation capacities. However, the high performance of CNN based edge detection is achieved with a large pretrained CNN backbone, which is memory and energy consuming. In addition, it is surprising that the previous wisdom from the traditional edge detectors, such as Canny, Sobel, and LBP are rarely investigated in the rapid-developing deep learning era. To address these issues, we propose a simple, lightweight yet effective architecture named Pixel Difference Network (PiDiNet) for efficient edge detection. Extensive experiments on BSDS500, NYUD, and Multicue are provided to demonstrate its effectiveness, and its high training and inference efficiency. Surprisingly, when training from scratch with only the BSDS500 and VOC datasets, PiDiNet can surpass the recorded result of human perception (0.807 vs. 0.803 in ODS F-measure) on the BSDS500 dataset with 100 FPS and less than 1M parameters. A faster version of PiDiNet with less than 0.1M parameters can still achieve comparable performance among state of the arts with 200 FPS. Results on the NYUD and Multicue datasets show similar observations. The codes are available at https://github.com/zhuoinoulu/pidinet.

Code Repositories

hellozhuo/pidinet
pytorch
Mentioned in GitHub
zhuoinoulu/pidinet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
edge-detection-on-brindPiDiNet
Number of parameters (M): 710K
ODS: 0.789
edge-detection-on-udedPiDiNet
ODS: 0.812

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp