AutoNet.h File Reference

Detailed Description

NVIDIA DriveWorks API: AutoNet Module

Description: This file defines the AutoNet module.

Definition in file AutoNet.h.

Go to the source code of this file.

Data Structures
struct	dwAutoNetParams
	AutoNet model parameters. More...

Typedefs
typedef struct dwAutoNetObject *	dwAutoNetHandle_t
	Handle to a AutoNet object. More...
typedef struct dwAutoNetOutputObject *	dwAutoNetOutputHandle_t
	Handle to a AutoNet output object. More...

Enumerations
enum	dwAutoNetBatchSize { DW_AUTONET_BATCH_SIZE_1 = 0 }
	AutoNet network batch size options. More...
enum	dwAutoNetHeadType {   DW_AUTONET_DRIVENET_HEAD = 0,   DW_AUTONET_MAPNET_HEAD = 1,   DW_AUTONET_OPENROADNET_HEAD = 2,   DW_AUTONET_PATHNET_HEAD = 3,   DW_AUTONET_WAITNET_HEAD = 4,   DW_AUTONET_URGENCYNET_HEAD = 5,   DW_AUTONET_LIGHTSOURCENET_HEAD = 6,   DW_AUTONET_HEAD_COUNT }
	Definition AutoNet DNN heads. More...
enum	dwAutoNetModel {   DW_AUTONET_MODEL_FRONT = 0,   DW_AUTONET_NUM_MODELS,   DW_AUTONET_MODEL_CUSTOM = -1 }
	AutoNet network models to be loaded by AutoNet module. More...
enum	dwAutoNetSelectionMask {   DW_AUTONET_SELECTION_MASK_DRIVENET = (1 << DW_AUTONET_DRIVENET_HEAD),   DW_AUTONET_SELECTION_MASK_MAPNET = (1 << DW_AUTONET_MAPNET_HEAD),   DW_AUTONET_SELECTION_MASK_OPENROADNET = (1 << DW_AUTONET_OPENROADNET_HEAD),   DW_AUTONET_SELECTION_MASK_PATHNET = (1 << DW_AUTONET_PATHNET_HEAD),   DW_AUTONET_SELECTION_MASK_WAITNET = (1 << DW_AUTONET_WAITNET_HEAD),   DW_AUTONET_SELECTION_MASK_URGENCYNET = (1 << DW_AUTONET_URGENCYNET_HEAD),   DW_AUTONET_SELECTION_MASK_LIGHTSOURCENET = (1 << DW_AUTONET_LIGHTSOURCENET_HEAD),   DW_DNN_SELECTION_MASK_ALL = (1 << DW_AUTONET_HEAD_COUNT) - 1 }
	Bitmask enum which enables specific DNN heads to be parsed and post-processed. More...

Functions
DW_API_PUBLIC dwStatus	dwAutoNet_bindTensorTimestamp (dwAutoNetOutputHandle_t autonetOutput, dwTime_t tensorTimeStamp)
	Ties image timestamp to the autonet output tensor. More...
DW_API_PUBLIC dwStatus	dwAutoNet_createOutput (dwAutoNetOutputHandle_t *autonetOutput, dwAutoNetHandle_t obj)
	Allocates AutoNet output. More...
DW_API_PUBLIC dwStatus	dwAutoNet_destroyOutput (dwAutoNetOutputHandle_t autonetOutput)
	Destroys AutoNet output. More...
DW_API_PUBLIC dwStatus	dwAutoNet_getDNNMetaData (dwDNNMetaData *metaData, dwAutoNetHandle_t obj)
	Returns the DNN metadata. More...
DW_API_PUBLIC dwStatus	dwAutoNet_getInputBlobsize (dwBlobSize *inputBlobsize, dwAutoNetHandle_t obj)
	Returns input blob size that AutoNet expects. More...
DW_API_PUBLIC dwStatus	dwAutoNet_getInputTensorProperties (dwDNNTensorProperties *inputProps, dwAutoNetHandle_t obj)
	Returns input properties that AutoNet expects. More...
DW_API_PUBLIC dwStatus	dwAutoNet_getNumHeadOutputBlobs (uint32_t *numOutputBlobs, dwAutoNetHeadType headType, dwAutoNetHandle_t obj)
	Returns number of valid head outputs that AutoNet expects. More...
DW_API_PUBLIC dwStatus	dwAutoNet_getOutputTensorProperties (dwDNNTensorProperties *outputProps, dwAutoNetHeadType headType, dwAutoNetHandle_t obj)
	Returns output properties that AutoNet expects. More...
DW_API_PUBLIC dwStatus	dwAutoNet_getProcessorType (dwProcessorType *processorType, dwAutoNetHandle_t obj)
	Returns processor type that AutoNet runs on. More...
DW_API_PUBLIC dwStatus	dwAutoNet_inferAsync (dwAutoNetOutputHandle_t autonetOutput, dwConstDNNTensorHandle_t *inputTensors, dwAutoNetHandle_t obj)
	Runs inference on given list of input tensors. More...
DW_API_PUBLIC dwStatus	dwAutoNet_initDefaultParams (dwAutoNetParams *params)
	Initializes AutoNet parameters with default values. More...
DW_API_PUBLIC dwStatus	dwAutoNet_initialize (dwAutoNetHandle_t *autoNetHandle, const dwAutoNetParams *autoNetParams, dwContextHandle_t ctx)
	Initializes AutoNet module. More...
DW_API_PUBLIC dwStatus	dwAutoNet_release (dwAutoNetHandle_t obj)
	Releases the AutoNet module. More...
DW_API_PUBLIC dwStatus	dwAutoNet_reset (dwAutoNetHandle_t obj)
	Resets AutoNet. More...
DW_API_PUBLIC dwStatus	dwAutoNet_setCUDAStream (cudaStream_t stream, dwAutoNetHandle_t obj)
	Sets the CUDA stream for CUDA related operations. More...