yolov8pose 部署rknn(rk3588)、部署地平线Horizon、部署TensorRT,部署工程难度小、模型推理速度快,DFL放后处理中
创始人
2024-11-10 15:12:12
0

  特别说明:参考官方开源的yolov8代码、瑞芯微官方文档、地平线的官方文档,如有侵权告知删,谢谢。

  模型和完整仿真测试代码,放在github上参考链接 模型和代码。

  之前写了yolov8、yolov8seg、yolov8obb 的 DFL 放在模型中和放在后处理中的两种不同部署方法(推荐放在后处理中);而yolov8pose只写了一篇放在模型中,有网友希望写一篇放在后处理中的,yolov8pose的DFL放在后处理中的博客来了。

1 模型和训练

  老规矩,训练不涉及,训练代码参考官方开源的yolov8训练代码。

2 导出 yolov8 onnx

  修改两处、增加保存onnx代码,一共修改三个地方。

修改第一处:增加以下几行代码
在这里插入图片描述

        # 导出 onnx 增加         y = []         for i in range(self.nl):             t1 = self.cv2[i](x[i])             t2 = self.cv3[i](x[i])             y.append(t1)             y.append(t2)         return y 

第二处:增加以下几行代码
在这里插入图片描述

        ps = []         for i in range(self.nl):             ps.append(self.cv4[i](x[i]))         x = self.detect(self, x)         return x, ps 

  增加保存onnx模型代码
在这里插入图片描述

        print("===========  onnx =========== ")         dummy_input = torch.randn(1, 3, 640, 640)         input_names = ["data"]         output_names = ["reg1", "cls1", "reg2", "cls2", "reg3", "cls3", "ps1", "ps2", "ps3"]         torch.onnx.export(self.model, dummy_input, "./yolov8n_pose.onnx", verbose=False, input_names=input_names, output_names=output_names, opset_version=11)         print("======================== convert onnx Finished! .... ") 

  修改完以上几个地方,运行以下两行。

from ultralytics import YOLO model = YOLO('./weights/yolov8n-pose.pt') 

3 yolov8n-pose onnx 测试效果

在这里插入图片描述

4 时耗

将DFL放后处理中,模型和后处理时耗,使用芯片rk3588,模型输入分辨率640x640。部署rk3588板端代码
在这里插入图片描述

以下截图是将DFL放在模型中的时耗(模型推理时会长一些,但后处理会稍微快一点),将DFL放在模型中部署【参考链接】

在这里插入图片描述

相关内容

热门资讯

微扑克ai机器人(微扑克)微扑... 微扑克ai机器人(微扑克)微扑克插件(透视)本来是真的有挂(详细辅助详细教程)1、微扑克ai机器人机...
wepoke是真的有挂(透视)... wepoke是真的有挂(透视)wepoke软件机器人(详细辅助透明教程)其实是有挂(详细有挂);1、...
微扑克ai机器人(微扑克)微扑... 微扑克ai机器人(微扑克)微扑克有辅助插件吗(透视)其实是有挂(详细辅助技巧教程)1、微扑克ai机器...
WePoKe透视挂(透视)we... WePoKe透视挂(透视)wepower透视辅助(详细辅助大神讲解)确实是真的有挂(专业外 挂)1、...
微扑克辅助软件(微扑克)微扑克... 微扑克辅助软件(微扑克)微扑克辅助工具(透视)都是有挂(详细辅助透视教程);1、构建自己的微扑克辅助...
wepoke黑科技(透视)we... wepoke黑科技(透视)wepoke黑科技是啥(详细辅助必胜教程)确实存在有挂(科普透视挂)1、w...
微扑克ai辅助工具(微扑克)w... 微扑克ai辅助工具(微扑克)wpk微扑克外挂事件(透视)真是有挂(详细辅助切实教程);1、微扑克ai...
wepoke插件(透视)wep... wepoke插件(透视)wepoke怎么使用模拟器(详细辅助新2025版)一直是真的有挂(教你外 挂...
wepoke确实有挂(透视)w... wepoke确实有挂(透视)wepoke ai代打(详细辅助攻略教程)确实是真的有挂(科普透明挂);...
微扑克辅助挂(微扑克)微扑克网... 微扑克辅助挂(微扑克)微扑克网页版辅助(透视)一直存在有挂(详细辅助辅助教程)1、玩家可以在微扑克辅...