1 Star 0 Fork 0

蓝网 | 程序设计 | 视觉组 / Ultralytics YOLOv8

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README
MIT

Ultralytics YOLOv8

介绍

快速上手YOLOv8并在streamlit上实践模型

官方快速学习通道

  • 因为上面链接讲的非常完备,足够让你quick start

数据收集-->预模型加载-->模型训练-->模型预测 全套大礼包

1.conda创建自己的虚拟环境

安装库

pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install roboflow -i https://pypi.tuna.tsinghua.edu.cn/simple

2.图片数据下载

这里搜索你想识别的任何图片数据集(同时也包括分割,分类等)

tips:这些数据集是开箱即用,都是已经做好标签和分好类的了

下载流程

点击download

点击download

点击continue

点击continue

点击复制然后粘贴到你的代码单元格上

点击复制

3.官方做法是在Colab(云主机,有限时的GPU资源)上实现,这对只有cpu电脑的人较友好,因为cpu根本跑不动

本地实现(使用ipynb):

导入库

from ultralytics import YOLO
import os
from IPython.display import display, Image
from IPython import display
display.clear_output()
!yolo mode=checks

然后粘贴上面的图片数据的下载代码,运行后就会将数据下载到当前目录,点击目录,修改data.yaml,将里面三个路径改为train,test,val的绝对路径(可以不改路径先训练,看看是否报错)

然后就是CLI式一段代码训练:

!yolo task=detect mode=train model=yolov8m.pt data={dataset.location}/data.yaml epochs=20 imgsz=640

这里训练完呢会在你的weights文件夹中生成一个best.pt,就是你训练得到的模型。

4.val & predict(可以跳过)

!yolo task=detect mode=val model=/path/to/pt data={dataset.location}/data.yaml

!yolo task=detect mode=predict model=/path/to/pt conf=0.5 data={dataset.location}/data.yaml

5.实例

!yolo task=detect mode=predict model=/content/runs/detect/train/weights/best.pt conf=0.5 source='path/to/your_pic_or_video'

打开外接摄像头进行实时识别,save是保存视频流,show是有视频流输出,跟cv2.imshow()一样

!yolo task=detect mode=predict model=RGB.pt conf=0.5 source=1 show=True save=True
MIT License Copyright (c) 2024 D-vision Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

简介

快速上手YOLOv8并在streamlit上实践模型 展开 收起
Python
MIT
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
Python
1
https://gitee.com/blue-net-vision/ultralytics-yolov8.git
git@gitee.com:blue-net-vision/ultralytics-yolov8.git
blue-net-vision
ultralytics-yolov8
Ultralytics YOLOv8
master

搜索帮助