文件名称：yolov5-onnx-tensorrt:如何使用 TensorRT 运行 yolov5 模型
文件大小：61KB
文件格式：ZIP
更新时间：2024-06-18 09:34:54
pytorch tensorrt onnx jetson-agx-xavier yolov5-model
yolov5-onnx-张量 此 Repos 包含如何使用 TensorRT 运行 yolov5 模型。 Pytorch 实现是 。 将 pytorch 转换为 onnx 和 tensorrt yolov5 模型以在 Jetson AGX Xavier 上运行。 支持推断图像。 支持同时推断多幅图像。 要求 请使用torch>=1.6.0 + onnx==1.8.0 + TensorRT 7.0.0.11 运行代码 代码结构 networks代码是网络demo代码在 Jetson AGX Xavier 上运行 tensorrt 实现 ├── networks ├── utils ├── models ├── demo │ ├── demo.py | ├── demo_batch.py | ├── Processor.py | ├── Processor_Batch.py
【文件预览】：
yolov5-onnx-tensorrt-master
----utils()
--------evolve.sh(601B)
--------general.py(53KB)
--------datasets.py(38KB)
--------__init__.py(0B)
--------google_utils.py(5KB)
--------activations.py(2KB)
--------torch_utils.py(9KB)
----demo()
--------onnx_tensorrt.py(2KB)
--------yolov5_onnx.py(3KB)
--------Visualizer.py(2KB)
--------Processor.py(11KB)
--------classes.py(1016B)
--------demo.py(3KB)
--------demo_batch.py(3KB)
--------__version__.py(23B)
--------Processor_Batch.py(14KB)
----setup.py(336B)
----README.md(3KB)
----networks()
--------experimental.py(6KB)
--------yolo.py(11KB)
--------common.py(4KB)
----.gitignore(21B)