scispace - formally typeset
Open AccessProceedings ArticleDOI

Soft-NMS — Improving Object Detection with One Line of Code

TLDR
Soft-NMS as mentioned in this paper decays the detection scores of all other objects as a continuous function of their overlap with M. As per the design of the algorithm, if an object lies within the predefined overlap threshold, it leads to a miss.
Abstract
Non-maximum suppression is an integral part of the object detection pipeline. First, it sorts all detection boxes on the basis of their scores. The detection box M with the maximum score is selected and all other detection boxes with a significant overlap (using a pre-defined threshold) with M are suppressed. This process is recursively applied on the remaining boxes. As per the design of the algorithm, if an object lies within the predefined overlap threshold, it leads to a miss. To this end, we propose Soft-NMS, an algorithm which decays the detection scores of all other objects as a continuous function of their overlap with M. Hence, no object is eliminated in this process. Soft-NMS obtains consistent improvements for the coco-style mAP metric on standard datasets like PASCAL VOC2007 (1.7% for both R-FCN and Faster-RCNN) and MS-COCO (1.3% for R-FCN and 1.1% for Faster-RCNN) by just changing the NMS algorithm without any additional hyper-parameters. Using Deformable-RFCN, Soft-NMS improves state-of-the-art in object detection from 39.8% to 40.9% with a single model. Further, the computational complexity of Soft-NMS is the same as traditional NMS and hence it can be efficiently implemented. Since Soft-NMS does not require any extra training and is simple to implement, it can be easily integrated into any object detection pipeline. Code for Soft-NMS is publicly available on GitHub http://bit.ly/2nJLNMu.

read more

Citations
More filters
Journal ArticleDOI

A Review and Comparative Study on Probabilistic Object Detection in Autonomous Driving

TL;DR: In this paper , a review and comparative study on existing probabilistic object detection methods for autonomous driving applications is presented, where the authors provide an overview of practical uncertainty estimation methods in deep learning, and then systematically survey existing methods and evaluation metrics.
Posted Content

Video Self-Stitching Graph Network for Temporal Action Localization

TL;DR: This paper proposes a multi-level cross-scale solution dubbed as video self-stitching graph network (VSGN), which improves the localization performance of short actions as well as achieving the state-of-the-art overall performance on THUMOS-14 and ActivityNet-v1.3.
Posted Content

Stitcher: Feedback-driven Data Provider for Object Detection

TL;DR: Stitcher is presented, a feedback-driven data provider, which aims to train object detectors in a balanced way, and steadily improves performance by a large margin in all settings, especially for small objects, with nearly no additional computation in both training and testing stages.
Journal ArticleDOI

An Anchor-Free Convolutional Neural Network for Real-Time Surgical Tool Detection in Robot-Assisted Surgery

TL;DR: An anchor-free convolutional neural network (CNN) architecture is proposed, a novel frame-by-frame method using a compact stacked hourglass network, which models the surgical tool as a single point: the center point of its bounding box, which truly realizes real-time surgical tool detection in RAS videos.
Book ChapterDOI

Retinal Abnormalities Recognition Using Regional Multitask Learning

TL;DR: A novel deep learning based method that can recognise 36 different retinal diseases with a single model is proposed that uses a region-specific multi-task recognition model by learning diseases affecting different regions of the retina with three sub-networks.
References
More filters
Proceedings ArticleDOI

Deep Residual Learning for Image Recognition

TL;DR: In this article, the authors proposed a residual learning framework to ease the training of networks that are substantially deeper than those used previously, which won the 1st place on the ILSVRC 2015 classification task.
Proceedings ArticleDOI

Histograms of oriented gradients for human detection

TL;DR: It is shown experimentally that grids of histograms of oriented gradient (HOG) descriptors significantly outperform existing feature sets for human detection, and the influence of each stage of the computation on performance is studied.
Journal ArticleDOI

A Computational Approach to Edge Detection

TL;DR: There is a natural uncertainty principle between detection and localization performance, which are the two main goals, and with this principle a single operator shape is derived which is optimal at any scale.
Proceedings ArticleDOI

You Only Look Once: Unified, Real-Time Object Detection

TL;DR: Compared to state-of-the-art detection systems, YOLO makes more localization errors but is less likely to predict false positives on background, and outperforms other detection methods, including DPM and R-CNN, when generalizing from natural images to other domains like artwork.
Proceedings ArticleDOI

Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation

TL;DR: RCNN as discussed by the authors combines CNNs with bottom-up region proposals to localize and segment objects, and when labeled training data is scarce, supervised pre-training for an auxiliary task, followed by domain-specific fine-tuning, yields a significant performance boost.