专栏名称: OpenCV学堂

一个致力于计算机视觉OpenCV原创技术传播的公众号！OpenCV计算机视觉与tensorflow深度学习相关算法原创文章分享、函数使用技巧、源码分析与讨论、，计算机视觉前沿技术介绍，技术专家经验分享，人才交流，学习交流。

使用YOLOv8 PT原始文件实现推理预测

OpenCV学堂 · 公众号 · 科技自媒体 · 2024-08-02 23:27

正文

点击上方蓝字关注我们

微信公众号： OpenCV学堂

关注获取更多计算机视觉与深度学习知识

前言

今天有个人发我一个YOLOv8分类的ONNX格式文件，跟我说测试效果很拉跨，但是直接用PT文件，通过命令行效果还错，让他万分郁闷。其实Pytorch模型是支持直接使用原始的PT文件推理的，无论模型来自Torchvision还是YOLOv5或者YOLOv8。使用PT文件推理都非常简单。

PT文件YOLOv8 分类推理

支持Top1跟Top5结果返回，使用非常方便。代码简洁明了。演示代码如下：

输出结果

PT文件YOLOv8 检测推理

对象检测支持返回Boxes信息与检测结果跟原始图像信息，两行代码即可搞定推理。演示代码如下：

PT推理好处

跟模型的评估验证结果高度一致，模型精度没有导出损失，效果好，另外方便支持GPU或者CPU切换，是Python开发环境下的首选。

OpenCV4系统化学习

深度学习系统化学习

推荐阅读

OpenCV4.8+YOLOv8对象检测C++推理演示

ZXING+OpenCV打造开源条码检测应用

攻略 | 学习深度学习只需要三个月的好方法

三行代码实现 TensorRT8.6 C++ 深度学习模型部署

请到「今天看啥」查看全文

推荐文章

烂板套利 · AI存储芯片，最正宗的10家公司（精选）

16 小时前

烂板套利 · AI存储芯片，最正宗的10家公司（精选）

16 小时前

广东药监 · 国家药监局信息中心主任陈锋调研广东药品智慧监管建设工作

19 小时前

广东药监 · 国家药监局信息中心主任陈锋调研广东药品智慧监管建设工作

19 小时前

李楠或kkk · 红魔的同学虽然慢但是没有放弃。不过好在黑神话也慢，资料片什么时-20250303160422

昨天

新闻大连 · 替大连郑重“辟谣”：这不是AI！

2 天前

新闻大连 · 替大连郑重“辟谣”：这不是AI！

2 天前

云头版 · 天翼云，引领级！

3 天前

经典微小说 · 为什么我们越来越难再爱上一个人？

8 年前

商务范 · 从大牌到平价款，这12个双肩包品牌好看又能装

7 年前

医学界 · PM2.5致死新研究，颠覆原来认知！

7 年前

十点读书 · 《红楼梦》：职场上，太把自己当回事，会死得很惨

7 年前

北极星电力网 · 电力下午茶（2018年2月6日）

7 年前