Skip to content

Latest commit

 

History

History
62 lines (47 loc) · 1.11 KB

README.MD

File metadata and controls

62 lines (47 loc) · 1.11 KB
Raw

With NMS Plugin

Generate trt File More Details

python export.py -o mdoel.onnx -e model.trt --end2end --v8 only for yolov8
  • YOLOv8
  • YOLOv7
  • YOLOv6
  • YOLOX
  • YOLOv5
  • YOLOv3

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

Generate trt File More Details

python export.py -o mdoel.onnx -e model.trt --end2end
  • YOLOv8
  • YOLOv7
  • YOLOv6
  • YOLOX
  • YOLOv5
  • YOLOv3

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