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
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