search

taifyang / yolo-inference

C++ and Python implementations of YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv9, YOLOv10, YOLOv11 inference.
46 stars 5 forks source link

推理结果异常 #47

Open shanhedian2017 opened 1 week ago

shanhedian2017 commented 1 week ago

目前测试发现yoloV11 cls的结果显示不太对，yoloV10 libtorch推理的时候，调用这个模型yolov10n_fp32_gpu.torchscript，会报错，我的libtorch版本是1.13.1 gpu版本

taifyang commented 1 week ago

目前测试发现yoloV11 cls的结果显示不太对，yoloV10 libtorch推理的时候，调用这个模型yolov10n_fp32_gpu.torchscript，会报错，我的libtorch版本是1.13.1 gpu版本

你用的是什么系统？yoloV11 cls用的模型是哪个？另外libtorch gpu报错，大概率是环境配置问题。

shanhedian2017 commented 1 week ago

我用的是win10 vs2017,libtorch cpu推理没有问题，gpu推理报错

熊鹰 @.***

 

------------------ 原始邮件 ------------------ 发件人: "taifyang/yolo-inference" @.>; 发送时间: 2024年11月13日(星期三) 下午3:26 @.>; @.**@.>; 主题: Re: [taifyang/yolo-inference] 推理结果异常 (Issue #47)

目前测试发现yoloV11 cls的结果显示不太对，yoloV10 libtorch推理的时候，调用这个模型yolov10n_fp32_gpu.torchscript，会报错，我的libtorch版本是1.13.1 gpu版本

你用的是什么系统？yoloV11 cls用的模型是哪个？另外libtorch gpu报错，大概率是环境配置问题。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

taifyang commented 1 week ago

我用的是win10 vs2017,libtorch cpu推理没有问题，gpu推理报错 熊鹰 @.   … ------------------ 原始邮件 ------------------ 发件人: "taifyang/yolo-inference" @.>; 发送时间: 2024年11月13日(星期三) 下午3:26 @.>; @*.**@*.>; 主题: Re: [taifyang/yolo-inference] 推理结果异常 (Issue #47) 目前测试发现yoloV11 cls的结果显示不太对，yoloV10 libtorch推理的时候，调用这个模型yolov10n_fp32_gpu.torchscript，会报错，我的libtorch版本是1.13.1 gpu版本 你用的是什么系统？yoloV11 cls用的模型是哪个？另外libtorch gpu报错，大概率是环境配置问题。 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.>

你可以尝试打印一下torch::cuda::is_available()是否为true，如果是False的话就是环境没配好。

shanhedian2017 commented 1 week ago

libtorch的环境没配好，OpenCV4.5测试发现问题，用的这个模型yolov5n_seg_fp32.onnx。 ![Uploading 微信图片_20241114104714.png…]()

taifyang commented 1 week ago

你可以尝试一下opencv4.9以上的版本，限于个人精力无法测试