org.bytedeco.tensorrt.nvinfer

Class INMSLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.INMSLayer

All Implemented Interfaces:

AutoCloseable

@Namespace(value="nvinfer1")
 @NoOffset
 @Properties(inherit=nvinfer.class)
public class INMSLayer
extends ILayer

\class INMSLayer \brief A non-maximum suppression layer in a network definition. The NMS algorithm iterates through a set of bounding boxes and their confidence scores, in decreasing order of score. Boxes are selected if their score is above a given threshold, and their intersection-over-union (IoU) with previously selected boxes is less than or equal to a given threshold. This layer implements NMS per batch item and per class. For each batch item, the ordering of candidate bounding boxes with the same score is unspecified. The layer has the following inputs, in order of input index: * Boxes contains the input bounding boxes. It is a linear tensor of type kFLOAT or kHALF. It has shape [batchSize, numInputBoundingBoxes, numClasses, 4] if the boxes are per class, or [batchSize, numInputBoundingBoxes, 4] if the same boxes are to be used for each class. * Scores contains the per-box scores. It is a linear tensor of the same type as Boxes. It has shape [batchSize, numInputBoundingBoxes, numClasses]. * MaxOutputBoxesPerClass is the maximum number of output boxes per batch item per class. It is a scalar (0D tensor) of type kINT32. * IoUThreshold is the maximum IoU for selected boxes. It is a scalar (0D tensor) of type kFLOAT in the range [0.0f, 1.0f]. It is an optional input with default 0.0f. * ScoreThreshold is the value that a box score must exceed in order to be selected. It is a scalar (0D tensor) of type kFLOAT. It is an optional input with default 0.0f. The layer has the following outputs, in order of output index: * SelectedIndices contains the indices of the selected boxes. It is a linear tensor of type kINT32. It has shape [NumOutputBoxes, 3]. Each row contains a (batchIndex, classIndex, boxIndex) tuple. The output boxes are sorted in order of increasing batchIndex and then in order of decreasing score within each batchIndex. For each batchIndex, the ordering of output boxes with the same score is unspecified. If MaxOutputBoxesPerClass is a constant input, the maximum number of output boxes is batchSize * numClasses * min(numInputBoundingBoxes, MaxOutputBoxesPerClass). Otherwise, the maximum number of output boxes is batchSize * numClasses * numInputBoundingBoxes. The maximum number of output boxes is used to determine the upper-bound on allocated memory for this output tensor. * NumOutputBoxes is the number of output boxes in SelectedIndices. It is a scalar (0D tensor) of type kINT32. \warning There is a hardware-dependent limit K such that only the K highest scoring boxes in each batch item will be considered for selection. The value of K is 2000 for SM 5.3 and 6.2 devices, and 5000 otherwise. \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
INMSLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
nvinfer.BoundingBoxFormat	getBoundingBoxFormat() \brief Get the bounding box format parameter for the layer.
int	getTopKBoxLimit() \brief Get the TopK box limit parameter for the layer.
void	setBoundingBoxFormat(int fmt)
void	setBoundingBoxFormat(nvinfer.BoundingBoxFormat fmt) \brief Set the bounding box format parameter for the layer.
void	setTopKBoxLimit(int _limit) \brief Set the TopK box limit parameter for the layer.

Methods inherited from class org.bytedeco.tensorrt.nvinfer.ILayer

getInput, getMetadata, getName, getNbInputs, getNbOutputs, getOutput, getOutputType, getPrecision, getType, outputTypeIsSet, precisionIsSet, resetOutputType, resetPrecision, setInput, setMetadata, setMetadata, setName, setName, setOutputType, setOutputType, setPrecision, setPrecision

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

INMSLayer

public INMSLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setBoundingBoxFormat

@NoException(value=true)
public void setBoundingBoxFormat(nvinfer.BoundingBoxFormat fmt)

\brief Set the bounding box format parameter for the layer. The default value for the bounding box format parameter is kCORNER_PAIRS.

See Also:

nvinfer.BoundingBoxFormat, getBoundingBoxFormat()

setBoundingBoxFormat

@NoException(value=true)
public void setBoundingBoxFormat(@Cast(value="nvinfer1::BoundingBoxFormat")
                                                          int fmt)

getBoundingBoxFormat

@NoException(value=true)
public nvinfer.BoundingBoxFormat getBoundingBoxFormat()

\brief Get the bounding box format parameter for the layer.

See Also:

nvinfer.BoundingBoxFormat, setBoundingBoxFormat()

setTopKBoxLimit

@NoException(value=true)
public void setTopKBoxLimit(int _limit)

\brief Set the TopK box limit parameter for the layer. The TopK box limit is the maximum number of filtered boxes considered for selection per batch item. The default value for the TopK box limit parameter is 2000 for SM 5.3 and 6.2 devices, and 5000 otherwise. The TopK box limit must be less than or equal to {2000 for SM 5.3 and 6.2 devices, 5000 otherwise}.

See Also:

getTopKBoxLimit()

getTopKBoxLimit

@NoException(value=true)
public int getTopKBoxLimit()

\brief Get the TopK box limit parameter for the layer.

See Also:

setTopKBoxLimit()