imagenet-custom.md

^{Back | Next | Contents
Image Recognition}

Loading Custom Models on Jetson

The imagenet-console and imagenet-camera programs that we used before also accept extended command line parameters for loading a custom model snapshot. Set the $NET variable below to the path to your extracted snapshot:

$ NET=networks/GoogleNet-ILSVRC12-subset

$ ./imagenet-console bird_0.jpg output_0.jpg \
--prototxt=$NET/deploy.prototxt \
--model=$NET/snapshot_iter_184080.caffemodel \
--labels=$NET/labels.txt \
--input_blob=data \
--output_blob=softmax

As before, the classification and confidence will be overlayed to the output image. When compared to the output of the original network, the retrained GoogleNet-12 makes similar classifications to the original GoogleNet-1000, except that now it outputs the meta-classes that we've retrained it with:

The extended command line parameters above also load custom classification models with imagenet-camera.

Next | Locating Object Coordinates using DetectNet
Back | Downloading Model Snapshots to Jetson

^{© 2016-2019 NVIDIA | Table of Contents}