YOLOv8 目标检测项目实操

一 yolov8 背景介绍

YOLOv8是一种尖端的、最先进的(SOTA)模型,建立在以前 YOLO 版本的成功基础上,并引入了新的特性和改进,以进一步提高性能和灵活性。YOLOv8被设计为快速、准确、易于使用,这使它成为一个很好的选择,用于范围广泛的目标检测和跟踪、实例分割、图像分类和姿势估计任务。

一个不错的参考:yolov8官方代码训练模型

github yolov8 官网地址 地址:

GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

 预训练模型在 coco 数据集的执行效果:

1.1 环境安装

pip install ultralytics

1.2 安装cuda

使用的是重新配置的一套环境,使用最新版的cuda 本机执行没有成功,使用11.8可以

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

二 数据集

本项目场景比较通用,使用的网上的公共数据集,下载网站:

Roboflow Universe: Open Source Computer Vision Community

这里选择数据集下载格式:

三 执行训练 

3.1 先把代码拉到本地

git clone https://github.com/ultralytics/ultralytics.git

3.2 使用demo文件做测试

from ultralytics import YOLO# Load a model
model = YOLO('yolov8n.yaml')  # build a new model from YAML
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # build from YAML and transfer weights# Train the model
results = model.train(data='coco128.yaml', epochs=100, imgsz=640)

3.3 wandb 显示bug

这个报错显示了你正在使用的代码试图使用 Wandb(Weights & Biases),但是没有配置正确的 API 密钥。

要解决这个问题,你需要提供正确的 Wandb API 密钥。你可以按照之前的建议在代码中调用 wandb.login(key="your_api_key") 方法来配置 API 密钥。确保将 "your_api_key" 替换为你自己生成的 API 密钥。

3.3.1 解决方法

命令行执行:

wandb offline

方法二,没测试不知道是否可行:

import wandb
wandb.disabled = True

3.4 调整配置参数

注意 线程的数据 设置不能是8,默认设为0,设为8给我卡住了。

3.5 下载预训练模型

根据需要和场景下载适合的预训练模型:

3.6 开始训练

from ultralytics import YOLO
import multiprocessing# Load a model
model = YOLO('yolov8m.yaml')  # build a new model from YAML
model = YOLO('./yolov8m.pt')  # load a pretrained model (recommended for training)
model = YOLO('yolov8m.yaml').load('./yolov8m.pt')  # build from YAML and transfer weights# results = model.train(data='./data.yaml', epochs=10, imgsz=640)if __name__ == '__main__':multiprocessing.freeze_support()# 这里放置你的主程序代码# 例如,调用 model.train() 函数results = model.train(data='./data.yaml', epochs=500, imgsz=640)

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/625744.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

三相交流电子负载的重要指标

三相交流电子负载是一种模拟实际负载的电子设备,主要用于测试电源、变频器、逆变器等电力电子设备的性能和稳定性。在选择和使用三相交流电子负载时,需要关注以下几个重要指标: 1. 额定功率:三相交流电子负载的额定功率是指其能够…

Adobe发布Acrobat AI助手,PDF互动新体验;OpenAI推出Batch批处理API,支持半价优惠

🦉 AI新闻 🚀 Adobe发布Acrobat AI助手,PDF互动新体验 摘要:Adobe公司近日宣布,继2月推出测试版后,Acrobat AI 助手已在Acrobat Reader、桌面和网页上全面推出。以每月4.99美元的附加订阅形式提供&#x…

PgSQL之WITH Queries/Statement

PostgreSQL WITH 子句 在 PostgreSQL 中,WITH 子句提供了一种编写辅助语句的方法,以便在更大的查询中使用。 WITH 子句有助于将复杂的大型查询分解为更简单的表单,便于阅读。这些语句通常称为通用表表达式(Common Table Express…

React + Ts + Vite + Antd 项目搭建

1、创建项目 npm create vite 项目名称 选择 react 选择 typescript 关闭严格模式 建议关闭严格模式,因为不能自动检测副作用,有意双重调用。将严格模式注释即可。 2、配置sass npm install sass 更换所有后缀css为sass vite.config.ts中注册全局样式 /…

函数模板(C++)

目录 一、介绍 二、注意事项 三、排序函数 1、交换函数模板 2、排序算法 3、打印函数 4、测试函数 四、普通函数与函数模板 区别 调用规则 五、模板局限性 六、类模板 类模板与函数模板区别 1、类模板没有自动类型推导使用方式 2、类模板在模板参数列表中可以有默认…

【分享 网络墙测试】检测当前网络是否能用于其他平台,速度检测

文章日期:2024.04.17 类型:软件分享 兼容:win10 / win11 文章全程已做去敏处理!!! 【需要做的可联系我】 AES解密处理(直接解密即可)(crypto-js.js 标准算法&#xff09…

Springboot+Vue项目-基于Java+MySQL的网上超市系统(附源码+演示视频+LW)

大家好!我是程序猿老A,感谢您阅读本文,欢迎一键三连哦。 💞当前专栏:Java毕业设计 精彩专栏推荐👇🏻👇🏻👇🏻 🎀 Python毕业设计 &…

c++11 标准模板(STL)本地化库 - 平面类别(std::codecvt) - 在字符编码间转换,包括 UTF-8、UTF-16、UTF-32 (四)

本地化库 本地环境设施包含字符分类和字符串校对、数值、货币及日期/时间格式化和分析,以及消息取得的国际化支持。本地环境设置控制流 I/O 、正则表达式库和 C 标准库的其他组件的行为。 平面类别 在字符编码间转换,包括 UTF-8、UTF-16、UTF-32 std::…

async await与Promise.all

async await async和await是当前一个请求结束之后,才能发起下一个请求。这样会造成请求的阻塞,从而影响页面的渲染速度。请求时间时所有请求时间的和。 如下:方法触发后,会发起四个请求,四个请求需要的时间为239 134…

软件用户使用说明书

软件用户使用说明书 一、概述 本说明书旨在为用户提供详细的软件使用指南,帮助用户快速了解软件功能、操作方法以及常见问题解决方案。通过本说明书,用户可以轻松掌握软件的基本操作,提高使用效率。 二、软件安装 下载软件安装包&#xf…

C语言入门第四天(数组)

一、C语言数组的基本语法 1.数组的定义 数组是 C 语言中的一种数据结构,用于存储一组具有相同数据类型的数据。数组中的每个元素可以通过一个索引(下标)来访问,索引从 0 开始,最大值为数组长度减 1。 2.定义语法格式 …

Java Maven项目推送到 Maven 中央仓库

准备阶段 namespace 域名认证 当需要在 sonatype 认证 com.xxx命名空间时,需要将 .xxx.com 配置域名解析。 记录类型:TXT 文本内容:验证的 key。 GPG 公私钥生成 GPG 下载地址:https://www.gnupg.org/download/index.html M…