vgg-ssd-tensorRT

Implement VGG-SSD tensor layer by using TensorRT API

Due to time, I did not complete the training of the model,you can train your own model by your datasets and generates wts weights using pthTowts.py

Excute

build and run

mkdir build
cd build
cmake ..
make

// Serialize the model and generate ssd_mobilenet.engine

  ./vgg-ssd-tensorrt -s

// Deserialize and generate the detection results test.jpg and so on.

  ./vgg-tensorrt -d ../samples

GitHub

View Github