Functions
nms::nms Namespace Reference

Functions

def cpu_nms_wrapper
def gpu_nms_wrapper
def nms
def py_nms_wrapper

Function Documentation

def nms.nms.cpu_nms_wrapper (   thresh)

Definition at line 12 of file nms.py.

def nms.nms.gpu_nms_wrapper (   thresh,
  device_id 
)

Definition at line 18 of file nms.py.

def nms.nms.nms (   dets,
  thresh 
)
greedily select boxes with high confidence and overlap with current maximum <= thresh
rule out overlap >= thresh
:param dets: [[x1, y1, x2, y2 score]]
:param thresh: retain overlap < thresh
:return: indexes to keep

Definition at line 24 of file nms.py.

def nms.nms.py_nms_wrapper (   thresh)

Definition at line 6 of file nms.py.



rail_object_detector
Author(s):
autogenerated on Sat Jun 8 2019 20:26:31