从零训练yolov8

1.收集数据
2.数据标注

pip install labelimg

3.划分数据集 0.2的验证机0.8的训练集

import os
from shutil import copyfile
from sys import exit
import random


source = r"D:\Data\imgs\screenc" + '\\'
target_train = r"D:\Data\imgs\datasets\mydata\images\train" + '\\'
target_valid = r"D:\Data\imgs\datasets\mydata\images\valid" + '\\'


source_txt = r"D:\Data\imgs\lable" + '\\'
target_txt_train = r"D:\Data\imgs\datasets\mydata\lables\train" + '\\'
target_txt_valid = r"D:\Data\imgs\datasets\mydata\lables\valid" + '\\'

def creat_dir(dirs):
    if not os.path.exists(dirs):
        os.makedirs(dirs)


creat_dir(target_train)
creat_dir(target_valid)
creat_dir(target_txt_train)
creat_dir(target_txt_valid)

# 显示在所设置路径下的所有图片, filename这里仅为文件的文件名,如1.jpg
for filename in os.listdir(source):
    if '.png' in filename:
        source_filename = source + filename  # 加一个根目录编程图片的路径
        if(random.random()>0.2):
            target_train_filename = target_train + filename  # 加一个根目录编程图片的路径
            copyfile(source_filename, target_train_filename)
            source_txt_filename=source_txt+filename.split(".")[0]+".txt"
            copyfile(source_txt_filename, target_txt_train+filename.split(".")[0]+".txt")
        else:
            copyfile(source_filename,  target_valid+ filename)
            copyfile(source_txt+filename.split(".")[0]+".txt",  target_txt_valid+filename.split(".")[0]+".txt")

准确率(精确率),召回率
假定100人就诊,12人真的生病,10人被医生判为生病,9人被进一步检查确认真的生病。
准确率:9/10=0.9
召回率 9/12=0.75
混淆矩阵
f1=92/(92+3+1)=0.81
稳健率

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
你好!要从配置 YOLOv8,你需要按照以下步骤进行操作: 1. 环境配置: - 安装 Python,并确保运行 `python --version` 命令能够正确显示安装的版本。 - 安装 CUDA 和 cuDNN(如果你的 GPU 支持)以加速训练和推理过程。 - 安装 PyTorch 和 torchvision 库,可以通过运行 `pip install torch torchvision` 命令进行安装。 2. 下载 YOLOv4 源代码: - 克隆 YOLOv4 代码库:`git clone https://github.com/AlexeyAB/darknet.git` - 切换到 darknet 文件夹:`cd darknet` - 编译源代码:`make` 3. 配置数据和模型: - 下载预训练权重文件:`wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights` - 将配置文件复制到当前目录:`cp cfg/yolov4.cfg yolov4_custom.cfg` 4. 进行自定义训练: - 准备自定义数据集,并将图片和标签放入特定的文件夹中。 - 修改配置文件 `yolov4_custom.cfg`,将类别数目 (`classes`) 设置为你数据集中的类别数目,并确保 `filters` 的值等于 `(类别数目 + 5) * 3`。 - 创建 `obj.names` 文件,其中包含你数据集中的类别名称,每行一个类别。 - 创建 `obj.data` 文件,并指定数据集的路径和类别数目等信息。 - 运行 `darknet.exe detector train obj.data yolov4_custom.cfg yolov4.weights -dont_show` 开始训练。 请记住,YOLOv4 是一个复杂的模型,训练时间可能很长,尤其是在没有 GPU 加速的情况下。确保你的系统满足硬件要求,并准备好足够的计算资源和时间。此外,还可以根据你的需求和数据集调整模型的参数和训练策略。希望这些步骤对你有所帮助!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值