^{Image Recognition}</p>

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.

Back | <b><a href="imagenet-snapshot.md">Downloading Model Snapshots to Jetson</a></p> </b><p align="center">^{© 2016-2019 NVIDIA |}<a href="../README.md#two-days-to-a-demo-digits">^{Table of Contents}</a></p>