|
virtual | ~depthNet () |
| Destroy. More...
|
|
template<typename T > |
bool | Process (T *image, uint32_t width, uint32_t height) |
| Compute the depth field from a monocular RGB/RGBA image. More...
|
|
bool | Process (void *input, uint32_t width, uint32_t height, imageFormat format) |
| Compute the depth field from a monocular RGB/RGBA image. More...
|
|
template<typename T1 , typename T2 > |
bool | Process (T1 *input, T2 *output, uint32_t width, uint32_t height, cudaColormapType colormap=COLORMAP_VIRIDIS_INVERTED, cudaFilterMode filter=FILTER_LINEAR) |
| Process an RGB/RGBA image and map the depth image with the specified colormap. More...
|
|
bool | Process (void *input, imageFormat input_format, void *output, imageFormat output_format, uint32_t width, uint32_t height, cudaColormapType colormap=COLORMAP_VIRIDIS_INVERTED, cudaFilterMode filter=FILTER_LINEAR) |
| Process an RGB/RGBA image and map the depth image with the specified colormap. More...
|
|
template<typename T1 , typename T2 > |
bool | Process (T1 *input, uint32_t input_width, uint32_t input_height, T2 *output, uint32_t output_width, uint32_t output_height, cudaColormapType colormap=COLORMAP_DEFAULT, cudaFilterMode filter=FILTER_LINEAR) |
| Process an RGB/RGBA image and map the depth image with the specified colormap. More...
|
|
bool | Process (void *input, uint32_t input_width, uint32_t input_height, imageFormat input_format, void *output, uint32_t output_width, uint32_t output_height, imageFormat output_format, cudaColormapType colormap=COLORMAP_DEFAULT, cudaFilterMode filter=FILTER_LINEAR) |
| Process an RGB/RGBA image and map the depth image with the specified colormap. More...
|
|
template<typename T > |
bool | Visualize (T *output, uint32_t width, uint32_t height, cudaColormapType colormap=COLORMAP_DEFAULT, cudaFilterMode filter=FILTER_LINEAR) |
| Visualize the raw depth field into a colorized RGB/RGBA depth map. More...
|
|
bool | Visualize (void *output, uint32_t width, uint32_t height, imageFormat format, cudaColormapType colormap=COLORMAP_DEFAULT, cudaFilterMode filter=FILTER_LINEAR) |
| Visualize the raw depth field into a colorized RGB/RGBA depth map. More...
|
|
float * | GetDepthField () const |
| Return the raw depth field. More...
|
|
uint32_t | GetDepthFieldWidth () const |
| Return the width of the depth field. More...
|
|
uint32_t | GetDepthFieldHeight () const |
| Return the height of the depth field. More...
|
|
bool | SavePointCloud (const char *filename) |
| Extract and save the point cloud to a PCD file (depth only). More...
|
|
bool | SavePointCloud (const char *filename, float *rgba, uint32_t width, uint32_t height) |
| Extract and save the point cloud to a PCD file (depth + RGB). More...
|
|
bool | SavePointCloud (const char *filename, float *rgba, uint32_t width, uint32_t height, const float2 &focalLength, const float2 &principalPoint) |
| Extract and save the point cloud to a PCD file (depth + RGB). More...
|
|
bool | SavePointCloud (const char *filename, float *rgba, uint32_t width, uint32_t height, const float intrinsicCalibration[3][3]) |
| Extract and save the point cloud to a PCD file (depth + RGB). More...
|
|
bool | SavePointCloud (const char *filename, float *rgba, uint32_t width, uint32_t height, const char *intrinsicCalibrationPath) |
| Extract and save the point cloud to a PCD file (depth + RGB). More...
|
|
virtual | ~tensorNet () |
| Destory. More...
|
|
bool | LoadNetwork (const char *prototxt, const char *model, const char *mean=NULL, const char *input_blob="data", const char *output_blob="prob", uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true, nvinfer1::IInt8Calibrator *calibrator=NULL, cudaStream_t stream=NULL) |
| Load a new network instance. More...
|
|
bool | LoadNetwork (const char *prototxt, const char *model, const char *mean, const char *input_blob, const std::vector< std::string > &output_blobs, uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true, nvinfer1::IInt8Calibrator *calibrator=NULL, cudaStream_t stream=NULL) |
| Load a new network instance with multiple output layers. More...
|
|
bool | LoadNetwork (const char *prototxt, const char *model, const char *mean, const std::vector< std::string > &input_blobs, const std::vector< std::string > &output_blobs, uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true, nvinfer1::IInt8Calibrator *calibrator=NULL, cudaStream_t stream=NULL) |
| Load a new network instance with multiple input layers. More...
|
|
bool | LoadNetwork (const char *prototxt, const char *model, const char *mean, const char *input_blob, const Dims3 &input_dims, const std::vector< std::string > &output_blobs, uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true, nvinfer1::IInt8Calibrator *calibrator=NULL, cudaStream_t stream=NULL) |
| Load a new network instance (this variant is used for UFF models) More...
|
|
bool | LoadNetwork (const char *prototxt, const char *model, const char *mean, const std::vector< std::string > &input_blobs, const std::vector< Dims3 > &input_dims, const std::vector< std::string > &output_blobs, uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true, nvinfer1::IInt8Calibrator *calibrator=NULL, cudaStream_t stream=NULL) |
| Load a new network instance with multiple input layers (used for UFF models) More...
|
|
bool | LoadEngine (const char *engine_filename, const std::vector< std::string > &input_blobs, const std::vector< std::string > &output_blobs, nvinfer1::IPluginFactory *pluginFactory=NULL, deviceType device=DEVICE_GPU, cudaStream_t stream=NULL) |
| Load a network instance from a serialized engine plan file. More...
|
|
bool | LoadEngine (char *engine_stream, size_t engine_size, const std::vector< std::string > &input_blobs, const std::vector< std::string > &output_blobs, nvinfer1::IPluginFactory *pluginFactory=NULL, deviceType device=DEVICE_GPU, cudaStream_t stream=NULL) |
| Load a network instance from a serialized engine plan file. More...
|
|
bool | LoadEngine (nvinfer1::ICudaEngine *engine, const std::vector< std::string > &input_blobs, const std::vector< std::string > &output_blobs, deviceType device=DEVICE_GPU, cudaStream_t stream=NULL) |
| Load network resources from an existing TensorRT engine instance. More...
|
|
bool | LoadEngine (const char *filename, char **stream, size_t *size) |
| Load a serialized engine plan file into memory. More...
|
|
void | EnableLayerProfiler () |
| Manually enable layer profiling times. More...
|
|
void | EnableDebug () |
| Manually enable debug messages and synchronization. More...
|
|
bool | AllowGPUFallback () const |
| Return true if GPU fallback is enabled. More...
|
|
deviceType | GetDevice () const |
| Retrieve the device being used for execution. More...
|
|
precisionType | GetPrecision () const |
| Retrieve the type of precision being used. More...
|
|
bool | IsPrecision (precisionType type) const |
| Check if a particular precision is being used. More...
|
|
cudaStream_t | GetStream () const |
| Retrieve the stream that the device is operating on. More...
|
|
cudaStream_t | CreateStream (bool nonBlocking=true) |
| Create and use a new stream for execution. More...
|
|
void | SetStream (cudaStream_t stream) |
| Set the stream that the device is operating on. More...
|
|
const char * | GetPrototxtPath () const |
| Retrieve the path to the network prototxt file. More...
|
|
const char * | GetModelPath () const |
| Retrieve the full path to model file, including the filename. More...
|
|
const char * | GetModelFilename () const |
| Retrieve the filename of the file, excluding the directory. More...
|
|
modelType | GetModelType () const |
| Retrieve the format of the network model. More...
|
|
bool | IsModelType (modelType type) const |
| Return true if the model is of the specified format. More...
|
|
uint32_t | GetInputLayers () const |
| Retrieve the number of input layers to the network. More...
|
|
uint32_t | GetOutputLayers () const |
| Retrieve the number of output layers to the network. More...
|
|
Dims3 | GetInputDims (uint32_t layer=0) const |
| Retrieve the dimensions of network input layer. More...
|
|
uint32_t | GetInputWidth (uint32_t layer=0) const |
| Retrieve the width of network input layer. More...
|
|
uint32_t | GetInputHeight (uint32_t layer=0) const |
| Retrieve the height of network input layer. More...
|
|
uint32_t | GetInputSize (uint32_t layer=0) const |
| Retrieve the size (in bytes) of network input layer. More...
|
|
float * | GetInputPtr (uint32_t layer=0) const |
| Get the CUDA pointer to the input layer's memory. More...
|
|
Dims3 | GetOutputDims (uint32_t layer=0) const |
| Retrieve the dimensions of network output layer. More...
|
|
uint32_t | GetOutputWidth (uint32_t layer=0) const |
| Retrieve the width of network output layer. More...
|
|
uint32_t | GetOutputHeight (uint32_t layer=0) const |
| Retrieve the height of network output layer. More...
|
|
uint32_t | GetOutputSize (uint32_t layer=0) const |
| Retrieve the size (in bytes) of network output layer. More...
|
|
float * | GetOutputPtr (uint32_t layer=0) const |
| Get the CUDA pointer to the output memory. More...
|
|
float | GetNetworkFPS () |
| Retrieve the network frames per second (FPS). More...
|
|
float | GetNetworkTime () |
| Retrieve the network runtime (in milliseconds). More...
|
|
const char * | GetNetworkName () const |
| Retrieve the network name (it's filename). More...
|
|
float2 | GetProfilerTime (profilerQuery query) |
| Retrieve the profiler runtime (in milliseconds). More...
|
|
float | GetProfilerTime (profilerQuery query, profilerDevice device) |
| Retrieve the profiler runtime (in milliseconds). More...
|
|
void | PrintProfilerTimes () |
| Print the profiler times (in millseconds). More...
|
|
|
static uint32_t | VisualizationFlagsFromStr (const char *str, uint32_t default_value=VISUALIZE_INPUT|VISUALIZE_DEPTH) |
| Parse a string of one of more VisualizationMode values. More...
|
|
static depthNet * | Create (const char *network="fcn-mobilenet", uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true) |
| Load a pre-trained model. More...
|
|
static depthNet * | Create (const char *model_path, const char *input=DEPTHNET_DEFAULT_INPUT, const char *output=DEPTHNET_DEFAULT_OUTPUT, uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true) |
| Load a new network instance. More...
|
|
static depthNet * | Create (const char *model_path, const char *input, const Dims3 &inputDims, const char *output, uint32_t maxBatchSize=DEFAULT_MAX_BATCH_SIZE, precisionType precision=TYPE_FASTEST, deviceType device=DEVICE_GPU, bool allowGPUFallback=true) |
| Load a custom network instance of a UFF model. More...
|
|
static depthNet * | Create (int argc, char **argv) |
| Load a new network instance by parsing the command line. More...
|
|
static depthNet * | Create (const commandLine &cmdLine) |
| Load a new network instance by parsing the command line. More...
|
|
static const char * | Usage () |
| Usage string for command line arguments to Create() More...
|
|
static bool | LoadClassLabels (const char *filename, std::vector< std::string > &descriptions, int expectedClasses=-1) |
| Load class descriptions from a label file. More...
|
|
static bool | LoadClassLabels (const char *filename, std::vector< std::string > &descriptions, std::vector< std::string > &synsets, int expectedClasses=-1) |
| Load class descriptions and synset strings from a label file. More...
|
|
static bool | LoadClassColors (const char *filename, float4 *colors, int expectedClasses, float defaultAlpha=255.0f) |
| Load class colors from a text file. More...
|
|
static bool | LoadClassColors (const char *filename, float4 **colors, int expectedClasses, float defaultAlpha=255.0f) |
| Load class colors from a text file. More...
|
|
static float4 | GenerateColor (uint32_t classID, float alpha=255.0f) |
| Procedurally generate a color for a given class index with the specified alpha value. More...
|
|
static precisionType | SelectPrecision (precisionType precision, deviceType device=DEVICE_GPU, bool allowInt8=true) |
| Resolve a desired precision to a specific one that's available. More...
|
|
static precisionType | FindFastestPrecision (deviceType device=DEVICE_GPU, bool allowInt8=true) |
| Determine the fastest native precision on a device. More...
|
|
static std::vector< precisionType > | DetectNativePrecisions (deviceType device=DEVICE_GPU) |
| Detect the precisions supported natively on a device. More...
|
|
static bool | DetectNativePrecision (const std::vector< precisionType > &nativeTypes, precisionType type) |
| Detect if a particular precision is supported natively. More...
|
|
static bool | DetectNativePrecision (precisionType precision, deviceType device=DEVICE_GPU) |
| Detect if a particular precision is supported natively. More...
|
|
Mono depth estimation from monocular images, using TensorRT.