南京做网站营销,成品网站怎么新建网页,自己做企业网站用哪个软件,网站建设发展方向特别说明#xff1a;参考官方开源的 yoloworld 代码、瑞芯微官方文档、地平线的官方文档#xff0c;如有侵权告知删#xff0c;谢谢。 模型和完整仿真测试代码#xff0c;放在github上参考链接 模型和代码。 yoloworld出来的有一段时间了#xff0c;还没有盘到板端上玩一玩… 特别说明参考官方开源的 yoloworld 代码、瑞芯微官方文档、地平线的官方文档如有侵权告知删谢谢。 模型和完整仿真测试代码放在github上参考链接 模型和代码。 yoloworld出来的有一段时间了还没有盘到板端上玩一玩不把这个给整落地工作都干不起劲。落地过程也是一波三折多次想放弃起早赶晚的抽时间干再试一次再试最后一次再试最最后一次 。。。yoloworld 环境MMDet搭建都折腾了三次转onnx也折腾了三次上rknn板子又是三次尝试rknn_toolkit2-1.3.0、rknn_toolkit2-1.6.0运行报错最终用的rknn_toolkit2-2.0.0运行成功都快折腾废了。
1 模型训练 训练就不多说也没有尝试过训练参考官方代码本示例直接用yolov8提供的yoloworld模型本想用腾讯提供的模型奈何搭建环境太折腾尝试过Huggingface导出的onnx模型奈何规避不掉板端不支持的操作进行板端部署。
1 导出 yolo_world onnx yolo_world_v2_s 尝试过用 huggingface 提供的demo导出onnx但其中有操作torch.einsum板端支持不了只能放弃作罢。用yoloworld 官方提供的导出onnx方式可以规避 torch.einsum 操作得到可以上板子的onnx模型。导出时进行调整导出模型示意图如下导出的类别使用的是80类。本示例对应提供的代码只用于适用六个输出头。
2 pytorch测试效果 onnx测试效果
3 tensorRT 时耗 模型yolo_world_v2_s导出类别80类输入分辨率是640x640转trt使用的fp16_mode显卡Tesla V100cuda_11.0。
5 rknn 板端C部署 模型yolo_world_v2_s导出类别80类输入分辨率是640x640芯片rk3588. 把在rk3588板子上测试的模特推理时耗和用C代码写的后处理时耗都给贴出来供大家参考。【rk3588的C代码参考链接】。
