首页 > 其他分享 >Win10环境下yolov8(ultralytics) 快速配置与测试

Win10环境下yolov8(ultralytics) 快速配置与测试

时间:2023-01-31 14:13:39浏览次数:53  
标签:pt ultralytics yolov8 yolov8n https Win10 model

win10下亲测有效!(如果想在tensorrt+cuda下部署yolov8,直接看第五5章)

一、win10下创建yolov8环境

# 注:python其他版本在win10下,可能有坑,我已经替你踩坑了,这里python3.9亲测有效
conda create -n yolov8 python=3.9 -y
conda activate yolov8
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple

二、推理图像

模型下载地址:

# download offical weights(".pt" file)
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8s.pt
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8m.pt
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8l.pt
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x.pt
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x6.pt

这里下载yolov8n为例子,原图图下图:

我们将图像和yolov8n.pt放到路径:d:/Data/,推理:

yolo predict model="d:/Data/yolov8n.pt" source="d:/Data/6406407.jpg"

效果如图:

三、训练

3.1 快速训练coco128数据集

在win10下,创建路径:D:\CodePython\yolov8,将这个5Mb的数据集下载并解压在目录,coco128数据集快速下载:https://share.weiyun.com/C0noWh5W

如下图:

 

 

 新建train.py文件,代码如下:、

from ultralytics import YOLO
 
# Load a model
# yaml会自动下载
model = YOLO("yolov8n.yaml")  # build a new model from scratch
model = YOLO("d:/Data/yolov8n.pt")  # load a pretrained model (recommended for training)
 
# Train the model
results = model.train(data="coco128.yaml", epochs=100, imgsz=640)

训练指令:

 python train.py

如下图训练状态:

3.2 预测

新建predict.py文件,代码如下:

from ultralytics import YOLO
 
# Load a model
model = YOLO("d:/Data/yolov8n.pt")  # load an official model
 
# Predict with the model
results = model("d:/Data/6406407.jpg")  # predict on an image

预测指令:

 python predict.py

如下图预测窗口打印: 

四、导出onnx

pip install onnx
yolo mode=export model="d:/Data/yolov8n.pt" format=onnx dynamic=True

五、yolov8的tensorrt部署加速

TensorRT-Alpha基于tensorrt+cuda c++实现模型end2end的gpu加速,支持win10、linux,在2023年已经更新模型:YOLOv8, YOLOv7, YOLOv6, YOLOv5, YOLOv4, YOLOv3, YOLOX, YOLOR,pphumanseg,u2net,EfficientDet。
Windows10教程正在制作,可以关注TensorRT-Alphahttps://github.com/FeiYull/TensorRT-Alpha

标签:pt,ultralytics,yolov8,yolov8n,https,Win10,model
From: https://www.cnblogs.com/feiyull/p/17078770.html

相关文章

  • win10中Oracle完全卸载
    1、停止所有服务win+R输入services.msc打开服务,停止所有Oracle服务。2、卸载Oracle3、清理注册表1)运行regedit,选择HKEY_LOCAL_MACHINE\SOFTWARE\ORACLE,按del键删......
  • win10如何设置一键关机休眠切换系统图标
    1.一键关机2.一键休眠创建快捷方式——一键休眠C:\Windows\System32\rundll32.exepowrprof.dll,SetSuspendStateHibemate3.切换任务图标https://www.kafan.cn/A......
  • win10启动后黑屏很久才进入登录界面
    问题描述电脑某次更新后启动速度变得非常慢,主要表现在黑屏很久才能进入登录界面,以前开机差不多十秒就完成了。网上搜索了各种解决办法,包括:禁用所有非Microsoft服务;关闭ULPS......
  • 备忘录——win10系统中SVNClient 连接超时
    问题:win10笔记本使用华为SVNClient连接超时解决方案1.设备管理器-->网络适配器-->SVNAdapterV1.0,右键选择禁用设备.2.进入C:\Windows\System32\drivers目录下,替......
  • win10配置 sshkey
    win10配置sshkey一:生成sshkey打开powershell 或者cmd输入命令:ssh-keygen-trsa1提示输入文件名[不输入,使用默认文件名,回车]2提示输入sshkey密码[以后使用......
  • 华为C8650USB连接win10无法访问内部储存空间
    C8650,MIUI-破落MIUI-2.3.7无tf卡时显示内部储存150MB不可用,显示SD卡18MB可卸载猜测是分区问题计算机管理-设备-AndroidAdapter显示此设备配置不正确(code1)磁盘空间不足......
  • WIN10 调出“搜索工具”
    问题描述:默认不出现“文件大小”等方式的搜索解决方法......
  • Win10系统 JDK环境变量配置
    (1)右击“此电脑”,点击属性,找到高级系统设置。(2)新建,添加变量名JAVA_HOME,变量值为之前自己jdk的安装路径C:\ProgramFiles\Java\jdk1.8.0_231(3)找到系统变量里变量为Path的,选......
  • win10 重装系统后没有睡眠选项
     001、win+i打开设置  002、点击系统  003、点击电源和睡眠  004、点击其他电源设置  005、  006、  007、选中休眠,然后保存修......
  • Win10 (64 位) 安装汇编环境 (MASM)
    参考: 1. Win10(64位)安装汇编环境(MASM)----dgw博客-博客园(cnblogs.com)2. win10(64位)安装汇编环境(masm)_Safforn的博客-CSDN博客_masm安装 ......