Caffe/Pytorch转为TensorRT 4.0的示例代码

【实例简介】
这个代码是安装TensorRT 4.0后自带的，功能也有限，所以自己在目录中搜索一下就能看到。所以先自己找找，找不到再下载。有些人可能不知道有这样的范例，工作碰上很麻烦。所以这里就打包上传。
【实例截图】
【核心代码】
quantum7_10744099
└── examples
├── caffe_to_trt
│   ├── caffe_mnist.py
│   ├── __init__.py
│   └── __pycache__
│   ├── caffe_mnist.cpython-35.pyc
│   └── __init__.cpython-35.pyc
├── custom_layers
│   ├── custom_layers.py
│   ├── __init__.py
│   ├── __pycache__
│   │   ├── custom_layers.cpython-35.pyc
│   │   └── __init__.cpython-35.pyc
│   └── tensorrtplugins
│   ├── interfaces
│   │   └── plugins.i
│   ├── MANIFEST.in
│   ├── __pycache__
│   │   └── setup.cpython-35.pyc
│   ├── README.md
│   ├── setup.cfg
│   ├── setup.py
│   ├── src
│   │   └── FullyConnected.h
│   └── tensorrtplugins
│   ├── __init__.py
│   └── __pycache__
│   └── __init__.cpython-35.pyc
├── googlenet.py
├── __init__.py
├── lite_examples
│   ├── caffe_lite_engine.py
│   ├── __init__.py
│   ├── plan_lite_engine.py
│   ├── __pycache__
│   │   ├── caffe_lite_engine.cpython-35.pyc
│   │   ├── __init__.cpython-35.pyc
│   │   ├── plan_lite_engine.cpython-35.pyc
│   │   ├── pyt_lite_engine.cpython-35.pyc
│   │   └── tf_lite_engine.cpython-35.pyc
│   ├── pyt_lite_engine.py
│   └── tf_lite_engine.py
├── mnist_api.py
├── onnx_mnist.py
├── __pycache__
│   ├── googlenet.cpython-35.pyc
│   ├── __init__.cpython-35.pyc
│   ├── mnist_api.cpython-35.pyc
│   ├── onnx_mnist.cpython-35.pyc
│   └── uff_mnist.cpython-35.pyc
├── pytorch_to_trt
│   ├── __init__.py
│   ├── mnist.py
│   ├── __pycache__
│   │   ├── __init__.cpython-35.pyc
│   │   ├── mnist.cpython-35.pyc
│   │   └── pytorch_to_trt.cpython-35.pyc
│   └── pytorch_to_trt.py
├── resnet_as_a_service
│   ├── __init__.py
│   ├── __pycache__
│   │   ├── __init__.cpython-35.pyc
│   │   └── resnet_as_a_service.cpython-35.pyc
│   ├── query.sh
│   └── resnet_as_a_service.py
├── sample_onnx
│   ├── __init__.py
│   ├── __pycache__
│   │   ├── __init__.cpython-35.pyc
│   │   └── sample_onnx.cpython-35.pyc
│   └── sample_onnx.py
├── tf_to_trt
│   ├── __init__.py
│   ├── lenet5.py
│   ├── __pycache__
│   │   ├── __init__.cpython-35.pyc
│   │   ├── lenet5.cpython-35.pyc
│   │   └── tf_to_trt.cpython-35.pyc
│   └── tf_to_trt.py
└── uff_mnist.py

22 directories, 58 files