Skip to content

Files

Latest commit

9467a8b · Aug 16, 2022

History

History
This branch is 1 commit ahead of, 24 commits behind Linaom1214/TensorRT-For-YOLO-Series:main.

cpp

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Aug 15, 2022
Aug 13, 2022
Aug 16, 2022

With NMS Plugin

Edit code for your model

auto in_dims = engine->getBindingDimensions(engine->getBindingIndex("image_arrays"));

run

cd end2end && mkdir build && build
cmake ..
make
./yolo -model_path  engine   -image_path xxx.jpg

Without NMS Plugin

Edit code for your model

static const int INPUT_W = 640;
static const int INPUT_H = 640;
const char* INPUT_BLOB_NAME = "image_arrays";
const char* OUTPUT_BLOB_NAME = "outputs";

run

cd norm && mkdir build && build
cmake ..
make
./yolo ../model_trt.engine -i ../*.jpg