Skip to content

wcc-cn/tensorrt_accelerate_model_python

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

nvidia environment

cuda toolkit 12.6 + cudnn 9.6.0 + tensorrt 10.10.0.31

Project Introduction

|
|---build.py //build model source file
|---detect.py // trt infer source file
|---readme.md // readme
|   
|---img
|       1.jpg // test img
|       
|---model
|       yolo11m.onnx // onnx model file
|       yolo11m.pt // yolo pt file
|       yolo11m_sim.engine // trt engine file
|       yolo11m_sim.onnx // simplify onnx model file
|       
|---output
        1.jpg // result of yolo pt model detection
        3.bmp // result of trt detection

About

Use Tensorrt to accelerate model reasoning, including converting pt to onnx, generating engine and loading engine for model reasoning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages