48 #ifndef DW_DNN_DATACONDITIONER_H_ 49 #define DW_DNN_DATACONDITIONER_H_ 153 uint32_t maxNumImages,
235 uint32_t numImages,
const dwRect* roi,
267 uint32_t numImages,
const dwRect* roi,
281 dwDataConditionerHandle_t obj);
299 dwDataConditionerHandle_t obj);
304 #endif // DW_DNN_DATACONDITIONER_H_ DW_API_PUBLIC dwStatus dwDataConditioner_setCUDAStream(cudaStream_t stream, dwDataConditionerHandle_t obj)
Sets the CUDA stream for CUDA related operations.
const dwImageCUDA * meanImage
Mean image to be subtracted.
NVIDIA DriveWorks API: Core Types
float float32_t
Specifies POD types.
struct dwImageObject * dwImageHandle_t
#define DW_MAX_IMAGE_PLANES
struct dwDataConditionerObject * dwDataConditionerHandle_t
Handle to a DataConditioner.
NVIDIA DriveWorks API: DNN Data types
NVIDIA DriveWorks API: Core Methods
NVIDIA DriveWorks API: Image Conversion and Streaming Functionality
DW_API_PUBLIC dwStatus dwDataConditioner_prepareData(dwDNNTensorHandle_t tensorOutput, const dwImageHandle_t *inputImages, uint32_t numImages, const dwRect *roi, cudaTextureAddressMode addressMode, dwDataConditionerHandle_t obj)
Runs the configured transformations on an image.
bool splitPlanes
Boolean indicating whether planes should be split. Default is true.
NVIDIA DriveWorks API: DNNTensor Structures and Methods
bool ignoreAspectRatio
Boolean indicating whether the aspect ratio of the input image should be kept (false) or the image sh...
DW_API_PUBLIC dwStatus dwDataConditioner_initializeFromTensorProperties(dwDataConditionerHandle_t *obj, const dwDNNTensorProperties *outputProperties, uint32_t maxNumImages, const dwDataConditionerParams *dataConditionerParams, cudaStream_t stream, dwContextHandle_t ctx)
Initializes a DataConditioner module.
DW_API_PUBLIC dwStatus dwDataConditioner_initialize(dwDataConditionerHandle_t *obj, const dwBlobSize *networkInputBlobSize, const dwDataConditionerParams *dataConditionerParams, cudaStream_t stream, dwContextHandle_t ctx)
Initializes a DataConditioner module.
dwStatus
Status definition.
float32_t stdev[DW_MAX_IMAGE_PLANES]
Standard deviation with range [0, 255].
DW_API_PUBLIC dwStatus dwDataConditioner_release(dwDataConditionerHandle_t obj)
Releases the DataConditioner module.
bool doPerPlaneMeanNormalization
Boolean indicating whether to perform per-plane mean normalization. Default false.
DW_API_PUBLIC dwStatus dwDataConditioner_outputPositionToInput(float32_t *outputX, float32_t *outputY, float32_t inputX, float32_t inputY, const dwRect *roi, dwDataConditionerHandle_t obj)
Computes the position of a point from the interpreted DNN output on the input image.
DW_API_PUBLIC dwStatus dwDataConditioner_getOutputSize(dwBlobSize *outputBlobSize, dwDataConditionerHandle_t obj)
Computes the output size based on the input size and the operations that have been added...
enum cudaTextureAddressMode cudaTextureAddressMode
float32_t meanValue[DW_MAX_IMAGE_PLANES]
Mean value to be subtracted. Range [0, 255]. Default is 0.
Specifies DNNTensor properties.
float32_t scaleCoefficient
Scale pixel intensities. Default is 1.0.
struct dwContextObject * dwContextHandle_t
Context handle.
struct dwDNNTensorObject * dwDNNTensorHandle_t
Handles representing Deep Neural Network interface.
DW_API_PUBLIC dwStatus dwDataConditioner_reset(dwDataConditionerHandle_t obj)
Resets the DataConditioner module.
DW_API_PUBLIC dwStatus dwDataConditioner_prepareDataRaw(float32_t *d_outputImage, const dwImageCUDA *const *inputImages, uint32_t numImages, const dwRect *roi, cudaTextureAddressMode addressMode, dwDataConditionerHandle_t obj)
Runs the configured transformations on an image.
DW_API_PUBLIC dwStatus dwDataConditioner_initParams(dwDataConditionerParams *dataConditionerParams)
Initializes DataConditioner parameters with default values.
DW_API_PUBLIC dwStatus dwDataConditioner_getCUDAStream(cudaStream_t *stream, dwDataConditionerHandle_t obj)
Gets the CUDA stream used by the data conditioner.