Yolo_v8的安装测试

前言

如何安装Python版本的Yolo,有一段时间不用了,Yolo的版本也在不断地发展,所以重新安装了运行了一下,记录了下来,供参考。

一、搭建环境

1.1、创建Pycharm工程

首先创建好一个空白的工程,如下图:

1.2、查看cuda的版本

可以,获知当前电脑安装的cuda版本是12.1。

​​​​​​​1.3、安装cuda版本的Pytorch

先安装mkl

再用已有文件安装cuda版本的Pytorch,如下图:

这个安装当中之所以出现,原有torch版本的卸载问题,是因为工程继承自系统的Python311,已经安装了cpu版本的torch的缘故,这个uninstall后并不影响后续的安装。

我们,可以查看安装后的情况,如下图:

​​​​​​​1.4、一个测试Pytorch的简单程序

说明,cuda版本的Pytorch已经完全可用了。

二、Yolov8的安装

2.1、参考链接

2024最新的YOLOv8安装配置全流程,人人都可以学会的图像识别技术指南-CSDN博客

2.2 、安装 ultralytics

直接输入:pip install ultralytics

但是,这样做的后果是,安装了最新版本的ultralytics,导致torch也要进行更新,于是它就自动这样做了,如下图:

然后,你看,这个torch就已经不支持cuda了,如下图:

不过,这个也不要紧(cpu版本的torch也能用,这不是本文的重点),我们继续:

查看当前的yolo版本,如下图:

三、下载源码开始测试

官网下载源码:https://github.com/ultralytics/ultralytics

如图:

其实,这个版本,已经不是Yolov8,而是Yolov11了。

(是Yolov8还是Yolov11主要决定的是模型文件,而非整体代码,整体代码而言应该是兼容Yolov8和Yolov11的)

先下载下来,然后复制到Pycharm工程中进行测试,如图:

3.1、先做一个预测测试

输入:yolo predict model=yolov8n.pt source='ultralytics/assets/bus.jpg' device=cpu

然后,我们查看这个预测的结果:

显然这个预测是符合我们预测要求的。

3.2、在线coco训练测试

采用在线下载coco包的方式进行测试:

yolo train model = yolov8n.pt data = coco128.yaml epochs = 10 imgsz = 640

训练结束,如下图:

我们可以查看训练的结果:

3.3、离线coco训练测试

将coco128.yaml中的内容进行了修改,删除了其中Download部分,

并将其中的文件和标签的路径修改如下:

并且,将coco集中images和labels复制到对应的位置,如图:

至此,可采用自定义的my_coco128.yaml进行离线训练了,

输入:yolo train model = yolov8n.pt data = my_coco128.yaml epochs = 10 imgsz = 640

运行完毕,如下图:

运行结果是一致的,如下图:

为什么要这样做呢?就是为了后续训练自己打标的图片做准备。

相关推荐
晚霞的不甘6 小时前
CANN 支持多模态大模型:Qwen-VL 与 LLaVA 的端侧部署实战
人工智能·神经网络·架构·开源·音视频
华玥作者12 小时前
[特殊字符] VitePress 对接 Algolia AI 问答(DocSearch + AI Search)完整实战(下)
前端·人工智能·ai
AAD5558889912 小时前
YOLO11-EfficientRepBiPAN载重汽车轮胎热成像检测与分类_3
人工智能·分类·数据挖掘
王建文go12 小时前
RAG(宠物健康AI)
人工智能·宠物·rag
ALINX技术博客13 小时前
【202601芯动态】全球 FPGA 异构热潮,ALINX 高性能异构新品预告
人工智能·fpga开发·gpu算力·fpga
易营宝13 小时前
多语言网站建设避坑指南:既要“数据同步”,又能“按市场个性化”,别踩这 5 个坑
大数据·人工智能
春日见13 小时前
vscode代码无法跳转
大数据·人工智能·深度学习·elasticsearch·搜索引擎
Drgfd14 小时前
真智能 vs 伪智能:天选 WE H7 Lite 用 AI 人脸识别 + 呼吸灯带,重新定义智能化充电桩
人工智能·智能充电桩·家用充电桩·充电桩推荐
好家伙VCC14 小时前
### WebRTC技术:实时通信的革新与实现####webRTC(Web Real-TimeComm
java·前端·python·webrtc
萤丰信息14 小时前
AI 筑基・生态共荣:智慧园区的价值重构与未来新途
大数据·运维·人工智能·科技·智慧城市·智慧园区