Creating DetectNet Model with DIGITS
TensorRT下载地址:https://developer.nvidia.com/nvidia-tensorrt-download
TesnsoRT介绍文档:https://devblogs.nvidia.com/tensorrt-3-faster-tensorflow-inference/
TensorRT开发者指南:http://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html
TensorRT样例代码:http://docs.nvidia.com/deeplearning/sdk/tensorrt-developer

本文介绍了如何使用TX2结合TensorRT及DIGITS创建DetectNet模型进行图像识别和检测。通过选择coco-dog数据集,设置训练参数如Adam优化器、学习率,并利用预训练的GoogleNet加速训练。训练完成后,展示了在DIGITS中进行模型测试和验证的过程。
最低0.47元/天 解锁文章
3586





