Yolov5-7.0训练教程-程序员宅基地

技术标签: YOLO  python  图像处理  人工智能  视觉检测  

源码链接:

GitHub - ultralytics/yolov5 at v7.0icon-default.png?t=N2N8https://github.com/ultralytics/yolov5/tree/v7.0

Yolov5-7.0项目路径:

/data/LP/yolov5-7.0

数据集路径:

/data/data_set/an_g_q

数据集中源图路径:

/data/data_set/an_g_q/images/train

数据集中标签路径:

/data/data_set/an_g_q/labels/train

数据集处理:

       实例分割部分,使用labelme数据标注工具标注,保存成json文件,然后将json文件格式转换成yolo所需的txt格式。

       批量转换脚本 json222txt.py

def convert_json_label_to_yolov_seg_label():
    import glob
    import numpy as np
    import json
    import cv2
    import os
    json_path = "D:/TL/yolov5-7.0/data/Dqi"
    json_files = glob.glob(json_path + "/*.json")
    for json_file in json_files:
        # if json_file != r"C:/Users/jianming_ge/Desktop/code/handle_dataset/water_street/223.json":
        #     continue
        print(json_file)
        f = open(json_file)
        json_info = json.load(f)
        # print(json_info.keys())
        img = cv2.imread(os.path.join(json_path, json_info["imagePath"]))
        height, width, _ = img.shape
        np_w_h = np.array([[width, height]], np.int32)
        txt_file = json_file.replace(".json", ".txt")
        f = open(txt_file, "a")
        for point_json in json_info["shapes"]:
            txt_content = ""
            np_points = np.array(point_json["points"], np.int32)
            norm_points = np_points / np_w_h
            norm_points_list = norm_points.tolist()
            txt_content += "3 " + " ".join([" ".join([str(cell[0]), str(cell[1])]) for cell in norm_points_list]) + "\n"
            f.write(txt_content)



if __name__ == '__main__':
    convert_json_label_to_yolov_seg_label()

    目标检测部分:使用labelimg数据标注工具标注,直接保存为yolo所需的txt格式

数据集处理好之后创建数据集配置文件,dte数据集对应的数据配置文件yolov5.yaml 路径:/data/LP/yolov5-7.0/data/yolov5s.yaml

dte.yaml文件中需要写的部分

train:为训练集的绝对路径

val:为验证集的绝对路径(没有划分训练集和验证集可以写跟上面一样的)

names:为数据集所标注的类别,顺序必须跟标注时的顺序相对应

预训练模型选择:去官网下载yolov5s.pt、yolov5m.pt、yolov5l.pt、yolov5x.pt模型

       地址:Releases · ultralytics/yolov5 · GitHub

模型区别:速度 s>m>l>x

检测精度:x>l>>s

模型大小:x>l>>s

       看情况选择合适的模型下载下来放到yolov5-7.0目录下

       根据所选的预训练模型设置训练模型的配置文件

       例 : 模型选择yolov5s/pt,打开yolov5-7.0目录下的models目录,找到对应的yolov5s.yaml文件

只需要修改nc,  nc所代表的是数据的类别个数。

环境创建:/data/LP/yolov5-7.0/README.md下面Documentation目录下查看训练所需要的python环境版本。在anaconda3中创建一个对应版本的空环境。返回到项目这中先选择使用这个环境,

找到/data/LP/yolov5-7.0/requirements.txt文件打开pycharm中的控制台控制台输入  pip install -r requirements.Txt对环境进行配置   

直接在anaconda3控制台激活创建好的环境在输入pip install -r requirements.Txt对环境进行配置也是一样的

调参:

    /data/LP/yolov5-7.0/train.Py文件下找 parse_opt()函数

       parse_opt()函数下

           --weighes:预训练权重文件,传入选择的模型路径

           --cfg:模型对应的配置文件,传入对应的模型配置文件路径

           --data:数据集配置文件的路径

           --epochs:训练轮次,默认300轮,按需求调整

           --batch-size:批次大小,作用就是一次往GPU当中放入多少图片,默认16。根据自己的显存决定,训练时显存占用越大效果越好,但是如果爆显存也是无法训练的,这个参数的调整在实际应用中按照2的幂次方来调整训练出的模型效果最佳,例如调为8  16  32  64  128  256

           --worker:数据装载时cpu使用的线程数,默认为8,看实际情况调整,内存大就调大一点内存小就调小一点,跟batch-size一样按照2的幂次方来调整效果最好,实际工程项目中一般看情况选择 8 16 32 64

           --imgsz:图片大小,默认640

           --rect:是否采用矩形训练,意思就是采用原图的宽高比不对图片进行填充,默认不采用。上面的imgsz是将读取到的图片填充为640*640的正方形了

           --nosave:不保存模型,默认False(保存)

           --optimizer:优化器,提供了SGDAdamAdamW 三种,单独说这三个优化器是AdamWAdam好,Adam SGD好,实际训练中,数据不同,采用模型不同,类别个数,样本的难易程度等一些情况,无法明确说那个优化器效果好,只能根据训练时候的实际效果来看。

           --cache:是否提前缓存图片到内存,以加快训练速度,默认不缓存,不推荐缓存,数据量小的时候还可以,数据量大反而会拖慢训练,

           --device:训练的设备,默认0,代表第一个GPU,如果有多张显卡的情况,可以选择多可,可以节省很多训练时间。

           --patience:提前停止训练,默认参数100,意思是100轮不更新就停止训练

           --label-smoothing:标签平滑,默认不增强,可以根据自己数据实际情况设置,如果要设置的话个人建议设置小一点0.10.05就差不多了。

           --project:训练结果保存路径

           --noautoanchor:参数进化,遗传算法调参

           其余超参数基本不需要调整,默认就可以。

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/m0_58946307/article/details/129855694

智能推荐

c# 调用c++ lib静态库_c#调用lib-程序员宅基地

文章浏览阅读2w次,点赞7次,收藏51次。四个步骤1.创建C++ Win32项目动态库dll 2.在Win32项目动态库中添加 外部依赖项 lib头文件和lib库3.导出C接口4.c#调用c++动态库开始你的表演...①创建一个空白的解决方案,在解决方案中添加 Visual C++ , Win32 项目空白解决方案的创建:添加Visual C++ , Win32 项目这......_c#调用lib

deepin/ubuntu安装苹方字体-程序员宅基地

文章浏览阅读4.6k次。苹方字体是苹果系统上的黑体,挺好看的。注重颜值的网站都会使用,例如知乎:font-family: -apple-system, BlinkMacSystemFont, Helvetica Neue, PingFang SC, Microsoft YaHei, Source Han Sans SC, Noto Sans CJK SC, W..._ubuntu pingfang

html表单常见操作汇总_html表单的处理程序有那些-程序员宅基地

文章浏览阅读159次。表单表单概述表单标签表单域按钮控件demo表单标签表单标签基本语法结构<form action="处理数据程序的url地址“ method=”get|post“ name="表单名称”></form><!--action,当提交表单时,向何处发送表单中的数据,地址可以是相对地址也可以是绝对地址--><!--method将表单中的数据传送给服务器处理,get方式直接显示在url地址中,数据可以被缓存,且长度有限制;而post方式数据隐藏传输,_html表单的处理程序有那些

PHP设置谷歌验证器(Google Authenticator)实现操作二步验证_php otp 验证器-程序员宅基地

文章浏览阅读1.2k次。使用说明:开启Google的登陆二步验证(即Google Authenticator服务)后用户登陆时需要输入额外由手机客户端生成的一次性密码。实现Google Authenticator功能需要服务器端和客户端的支持。服务器端负责密钥的生成、验证一次性密码是否正确。客户端记录密钥后生成一次性密码。下载谷歌验证类库文件放到项目合适位置(我这边放在项目Vender下面)https://github.com/PHPGangsta/GoogleAuthenticatorPHP代码示例://引入谷_php otp 验证器

【Python】matplotlib.plot画图横坐标混乱及间隔处理_matplotlib更改横轴间距-程序员宅基地

文章浏览阅读4.3k次,点赞5次,收藏11次。matplotlib.plot画图横坐标混乱及间隔处理_matplotlib更改横轴间距

docker — 容器存储_docker 保存容器-程序员宅基地

文章浏览阅读2.2k次。①Storage driver 处理各镜像层及容器层的处理细节,实现了多层数据的堆叠,为用户 提供了多层数据合并后的统一视图②所有 Storage driver 都使用可堆叠图像层和写时复制(CoW)策略③docker info 命令可查看当系统上的 storage driver主要用于测试目的,不建议用于生成环境。_docker 保存容器

随便推点

网络拓扑结构_网络拓扑csdn-程序员宅基地

文章浏览阅读834次,点赞27次,收藏13次。网络拓扑结构是指计算机网络中各组件(如计算机、服务器、打印机、路由器、交换机等设备)及其连接线路在物理布局或逻辑构型上的排列形式。这种布局不仅描述了设备间的实际物理连接方式,也决定了数据在网络中流动的路径和方式。不同的网络拓扑结构影响着网络的性能、可靠性、可扩展性及管理维护的难易程度。_网络拓扑csdn

JS重写Date函数,兼容IOS系统_date.prototype 将所有 ios-程序员宅基地

文章浏览阅读1.8k次,点赞5次,收藏8次。IOS系统Date的坑要创建一个指定时间的new Date对象时,通常的做法是:new Date("2020-09-21 11:11:00")这行代码在 PC 端和安卓端都是正常的,而在 iOS 端则会提示 Invalid Date 无效日期。在IOS年月日中间的横岗许换成斜杠,也就是new Date("2020/09/21 11:11:00")通常为了兼容IOS的这个坑,需要做一些额外的特殊处理,笔者在开发的时候经常会忘了兼容IOS系统。所以就想试着重写Date函数,一劳永逸,避免每次ne_date.prototype 将所有 ios

如何将EXCEL表导入plsql数据库中-程序员宅基地

文章浏览阅读5.3k次。方法一:用PLSQL Developer工具。 1 在PLSQL Developer的sql window里输入select * from test for update; 2 按F8执行 3 打开锁, 再按一下加号. 鼠标点到第一列的列头,使全列成选中状态,然后粘贴,最后commit提交即可。(前提..._excel导入pl/sql

Git常用命令速查手册-程序员宅基地

文章浏览阅读83次。Git常用命令速查手册1、初始化仓库git init2、将文件添加到仓库git add 文件名 # 将工作区的某个文件添加到暂存区 git add -u # 添加所有被tracked文件中被修改或删除的文件信息到暂存区,不处理untracked的文件git add -A # 添加所有被tracked文件中被修改或删除的文件信息到暂存区,包括untracked的文件...

分享119个ASP.NET源码总有一个是你想要的_千博二手车源码v2023 build 1120-程序员宅基地

文章浏览阅读202次。分享119个ASP.NET源码总有一个是你想要的_千博二手车源码v2023 build 1120

【C++缺省函数】 空类默认产生的6个类成员函数_空类默认产生哪些类成员函数-程序员宅基地

文章浏览阅读1.8k次。版权声明:转载请注明出处 http://blog.csdn.net/irean_lau。目录(?)[+]1、缺省构造函数。2、缺省拷贝构造函数。3、 缺省析构函数。4、缺省赋值运算符。5、缺省取址运算符。6、 缺省取址运算符 const。[cpp] view plain copy_空类默认产生哪些类成员函数

推荐文章

热门文章

相关标签