资源算法TensorRT-SSD

TensorRT-SSD

2020-01-09 | |  35 |   0 |   0

TensorRT-SSD

Use TensorRT API to implement Caffe-SSD, SSD(channel pruning), Mobilenet-SSD

============================================ I hope my code will help you learn and understand the TensorRT API better. It’s welcome to discuss the deep learning algorithm, model optimization, TensorRT API and so on, and learn from each other.

#Introduction:

  1. The original Caffe-SSD can run 3-5fps on my jetson tx2.

  2. TensorRT-SSD can run 8-10fps on my jetson tx2.

  3. TensorRT-SSD(channel pruning) can run 16-17fps on my jetson tx2.

  4. TensorRT-Mobilenet-SSD can run 40-43fps on my jetson tx2(it‘s cool!), and run 100+fps on gtx1060.

#Requirements:

  1. TensorRT3.0

  2. Cuda8.0 or Cuda9.0

  3. OpenCV

The code will be published shortly...

==============================================

In the Other_layer_tensorRT folder, there are the implementation of some other layers with TensorRT api, including:

  1. PReLU

Continuously updated...

  1. 2018/02/06, update detection_out layer

  2. 2018/03/07, add the common.cpp file

  3. 2018/04/21, TensorFlow 1.7 wheel with JetPack 3.2.(enable TensorRT support)

    Python2.7:https://nvidia.app.box.com/v/TF170-py27-wTRT

    Python3.5:https://nvidia.app.box.com/v/TF170-py35-wTRT

  4. 2018/05/07, TensorRT parse two(many) models, see sample_parse_two_models.txt

  5. 2018/05/30, add MobileNet-SSD_iplugin.prototxt (21 classes)

  6. 2018/07/19, fix the error of Concat layer in pluginIplement.cpp


上一篇:mxnet-ssd

下一篇:SSD-Windows

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...