在jetson thor上高效部署π0.5和yolo模型

在thor上部署π0.5

有2种方法:

  1. 直接使用pytorch + torch.compile
  2. 使用pytorch + tensorrt engine

其中方法2的难点是如何将tensorrt engine与pytorch流程结合起来,

如使用pytorch + action DIT engine或者使用pytorch + llm tensorrt engine + action DIT engine

基本思路是将tensorrt engine调用封装成forward方法并动态替换pytorch实现,这样可以灵活地替换我们想优化性能的部分。再保持高灵活性的同时兼顾推理性能。

具体细节有个生产级的开源项目, 已经完成了量化,编译,部署。十分值得学习和参考:
https://github.com/happyAnger66-an/model_optimizer

相关推荐
AI棒棒牛13 小时前
YOLOv13最新创新改进系列:YOLOv13特征可视化,特征提取图,科技感满满,丰富实验神器!!!
人工智能·科技·yolo·目标检测·计算机视觉
钓了猫的鱼儿13 小时前
【数据集】红外无人机目标检测数据集VOC+YOLO格式10000张1类别
yolo·目标检测·无人机
輕華14 小时前
YOLOv5 实战:从 GitHub 拉取到自定义数据集训练
yolo·github
盼小辉丶15 小时前
PyTorch强化学习实战(6)——交叉熵方法详解与实现
人工智能·pytorch·python·强化学习
ZhengEnCi15 小时前
06-多头注意力机制 🎯
人工智能·pytorch·python
guo_xiao_xiao_16 小时前
YOLOv11城市道路自行车目标检测数据集-999张-Bicycle-1
人工智能·yolo·目标检测
guo_xiao_xiao_16 小时前
YOLOv11机场跑道大型客机-飞机目标检测数据集-201张-Airplane-1_2_2
人工智能·yolo·目标检测
羊羊小栈17 小时前
基于「YOLO目标检测 + 多模态AI分析」的宠物猫狗健康智能检测分析预警系统
人工智能·yolo·目标检测·计算机视觉·毕业设计·大作业
南斯拉夫的铁托18 小时前
YOLO学习笔记
笔记·学习·yolo
钓了猫的鱼儿18 小时前
【数据集】红外无人机交通车辆目标检测数据集VOC+YOLO格式2371张4类别
yolo·目标检测·无人机