search

chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD， SSD（channel pruning）， Mobilenet-SSD
250 stars 84 forks source link

DRIVEPX2 porting #12

Open sankim90 opened 6 years ago

sankim90 commented 6 years ago

Hi I'm very interested in your work.

I will try with Drive PX2.

I also want to convert the SSD algorithm to the Tensorrt API.

How to start?

I am reading the tensor rt document. Is this enough?

Do I have to do CUDA programming?

Do you need CUDA programming for your implementation?

chenzhi1992 commented 6 years ago
  1. You should see some tensor samples, for example the samplePlugin folder， the sampleFasterRCNN folder.
  2. If you want to implement the plugin layer, you have to do CUDA programming.
sankim90 commented 6 years ago

thanks a lot

How long did development take?

chenzhi1992 commented 6 years ago

2-6 months