Skip to content

Latest commit

 

History

History
35 lines (27 loc) · 592 Bytes

README.MD

File metadata and controls

35 lines (27 loc) · 592 Bytes

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