Jetson Inference
DNN Vision Library

Image recognition DNN (GoogleNet, AlexNet, ResNet) More...

Classes

class  imageNet
 Image recognition with classification networks, using TensorRT. More...
 

Macros

#define DETECTNET_USAGE_STRING
 Standard command-line options able to be passed to detectNet::Create() More...
 
#define IMAGENET_DEFAULT_INPUT   "data"
 Name of default input blob for imageNet model. More...
 
#define IMAGENET_DEFAULT_OUTPUT   "prob"
 Name of default output confidence values for imageNet model. More...
 
#define IMAGENET_USAGE_STRING
 Standard command-line options able to be passed to imageNet::Create() More...
 
#define POSENET_USAGE_STRING
 Standard command-line options able to be passed to poseNet::Create() More...
 

Detailed Description

Image recognition DNN (GoogleNet, AlexNet, ResNet)

Macro Definition Documentation

◆ DETECTNET_USAGE_STRING

#define DETECTNET_USAGE_STRING
Value:
"detectNet arguments: \n" \
" --network=NETWORK pre-trained model to load, one of the following:\n" \
" * ssd-mobilenet-v1\n" \
" * ssd-mobilenet-v2 (default)\n" \
" * ssd-inception-v2\n" \
" * pednet\n" \
" * multiped\n" \
" * facenet\n" \
" * coco-airplane\n" \
" * coco-bottle\n" \
" * coco-chair\n" \
" * coco-dog\n" \
" --model=MODEL path to custom model to load (caffemodel, uff, or onnx)\n" \
" --prototxt=PROTOTXT path to custom prototxt to load (for .caffemodel only)\n" \
" --labels=LABELS path to text file containing the labels for each class\n" \
" --input-blob=INPUT name of the input layer (default is '" DETECTNET_DEFAULT_INPUT "')\n" \
" --output-cvg=COVERAGE name of the coverge output layer (default is '" DETECTNET_DEFAULT_COVERAGE "')\n" \
" --output-bbox=BOXES name of the bounding output layer (default is '" DETECTNET_DEFAULT_BBOX "')\n" \
" --mean-pixel=PIXEL mean pixel value to subtract from input (default is 0.0)\n" \
" --batch-size=BATCH maximum batch size (default is 1)\n" \
" --threshold=THRESHOLD minimum threshold for detection (default is 0.5)\n" \
" --alpha=ALPHA overlay alpha blending value, range 0-255 (default: 120)\n" \
" --overlay=OVERLAY detection overlay flags (e.g. --overlay=box,labels,conf)\n" \
" valid combinations are: 'box', 'labels', 'conf', 'none'\n" \
" --profile enable layer profiling in TensorRT\n\n"
#define DETECTNET_DEFAULT_BBOX
Name of default output blob of the grid of bounding boxes for DetectNet caffe model.
Definition: detectNet.h:46
#define DETECTNET_DEFAULT_COVERAGE
Name of default output blob of the coverage map for DetectNet caffe model.
Definition: detectNet.h:40
#define DETECTNET_DEFAULT_INPUT
Name of default input blob for DetectNet caffe model.
Definition: detectNet.h:34

Standard command-line options able to be passed to detectNet::Create()

◆ IMAGENET_DEFAULT_INPUT

#define IMAGENET_DEFAULT_INPUT   "data"

Name of default input blob for imageNet model.

◆ IMAGENET_DEFAULT_OUTPUT

#define IMAGENET_DEFAULT_OUTPUT   "prob"

Name of default output confidence values for imageNet model.

◆ IMAGENET_USAGE_STRING

#define IMAGENET_USAGE_STRING
Value:
"imageNet arguments: \n" \
" --network=NETWORK pre-trained model to load, one of the following:\n" \
" * alexnet\n" \
" * googlenet (default)\n" \
" * googlenet-12\n" \
" * resnet-18\n" \
" * resnet-50\n" \
" * resnet-101\n" \
" * resnet-152\n" \
" * vgg-16\n" \
" * vgg-19\n" \
" * inception-v4\n" \
" --model=MODEL path to custom model to load (caffemodel, uff, or onnx)\n" \
" --prototxt=PROTOTXT path to custom prototxt to load (for .caffemodel only)\n" \
" --labels=LABELS path to text file containing the labels for each class\n" \
" --input-blob=INPUT name of the input layer (default is '" IMAGENET_DEFAULT_INPUT "')\n" \
" --output-blob=OUTPUT name of the output layer (default is '" IMAGENET_DEFAULT_OUTPUT "')\n" \
" --batch-size=BATCH maximum batch size (default is 1)\n" \
" --profile enable layer profiling in TensorRT\n\n"
#define IMAGENET_DEFAULT_INPUT
Name of default input blob for imageNet model.
Definition: imageNet.h:34
#define IMAGENET_DEFAULT_OUTPUT
Name of default output confidence values for imageNet model.
Definition: imageNet.h:40

Standard command-line options able to be passed to imageNet::Create()

◆ POSENET_USAGE_STRING

#define POSENET_USAGE_STRING
Value:
"poseNet arguments: \n" \
" --network=NETWORK pre-trained model to load, one of the following:\n" \
" * resnet18-body (default)\n" \
" * resnet18-hand\n" \
" * densenet121-body\n" \
" --model=MODEL path to custom model to load (caffemodel, uff, or onnx)\n" \
" --prototxt=PROTOTXT path to custom prototxt to load (for .caffemodel only)\n" \
" --labels=LABELS path to text file containing the labels for each class\n" \
" --input-blob=INPUT name of the input layer (default is '" POSENET_DEFAULT_INPUT "')\n" \
" --output-cvg=COVERAGE name of the coverge output layer (default is '" POSENET_DEFAULT_CMAP "')\n" \
" --output-bbox=BOXES name of the bounding output layer (default is '" POSENET_DEFAULT_PAF "')\n" \
" --mean-pixel=PIXEL mean pixel value to subtract from input (default is 0.0)\n" \
" --batch-size=BATCH maximum batch size (default is 1)\n" \
" --threshold=THRESHOLD minimum threshold for detection (default is 0.5)\n" \
" --overlay=OVERLAY detection overlay flags (e.g. --overlay=links,keypoints)\n" \
" valid combinations are: 'box', 'links', 'keypoints', 'none'\n" \
" --keypoint-scale=X radius scale for keypoints, relative to image (default: 0.0052)\n" \
" --link-scale=X line width scale for links, relative to image (default: 0.0013)\n" \
" --profile enable layer profiling in TensorRT\n\n"
#define POSENET_DEFAULT_INPUT
Name of default input blob for pose estimation ONNX model.
Definition: poseNet.h:35
#define POSENET_DEFAULT_CMAP
Name of default output blob of the confidence map for pose estimation ONNX model. ...
Definition: poseNet.h:41
#define POSENET_DEFAULT_PAF
Name of default output blob of the Part Affinity Field (PAF) for pose estimation ONNX model...
Definition: poseNet.h:47

Standard command-line options able to be passed to poseNet::Create()