class Nms extends Serializable
Non-Maximum Suppression (nms) for Object Detection The goal of nms is to solve the problem that groups of several detections near the real location, ideally obtaining only one detection per object
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- Nms
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Instance Constructors
- new Nms()
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
def
clone(): AnyRef
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
finalize(): Unit
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- def isKeepCurIndex(boxArray: Array[Float], offset: Int, rowLength: Int, areas: Array[Float], curInd: Int, adaptiveThresh: Float, indices: Array[Int], indexLength: Int, normalized: Boolean): Boolean
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
nms(scores: Tensor[Float], boxes: Tensor[Float], thresh: Float, indices: Array[Int], sorted: Boolean = false, orderWithBBox: Boolean = false): Int
1.
1. first sort the scores from highest to lowest and get indices 2. for the bbox of first index, get the overlap between this box and the remaining bboxes put the first index to result buffer 3. update the indices by keeping those bboxes with overlap less than thresh 4. repeat 2 and 3 until the indices are empty
- scores
score tensor
- boxes
box tensor, with the size N*4
- thresh
overlap thresh
- indices
buffer to store indices after nms
- sorted
whether the scores are sorted
- orderWithBBox
whether return indices in the order with original bbox index
- returns
the length of indices after nms
- def nmsFast(scores: Tensor[Float], boxes: Tensor[Float], nmsThresh: Float, scoreThresh: Float, indices: Array[Int], topk: Int = -1, eta: Float = 1, normalized: Boolean = true): Int
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
def
toString(): String
- Definition Classes
- AnyRef → Any
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )