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
1