YOLOV7运行步骤(推理、训练全过程)

news/2024/7/8 4:12:35
  1. 下载源代码:点击下载

  2. 执行以下命令安装requirements.txt中的相关依赖

    pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
    
  3. 官网下载权重yolov7.pt(测试使用)、yolov7-tiny.pt(训练使用,这里使用什么模型则使用对应的权重)

  4. 在项目根目录下建立文件夹weights,把刚刚下载好的权重放进去

  5. 检测(detect)

    python detect.py --weights weights/yolov7.pt --source inference/images --device 0
    
  6. 训练(training)

    1. 准备YOLO格式的数据集,然后将数据集放在根目录下

      在这里插入图片描述

    2. 修改配置文件

      1. 在data文件夹下新建立一个xxx.yaml文件(这里我建立的是voc.yaml),然后将同目录下的coco.yaml文件内容复制进去,并修改以下内容,train、val、test分别为训练、验证、测试图片路径所在的文件,nc代表类别数 ,这里我使用的VOC2007,所以为20类,names则为各个,类别的名称

        在这里插入图片描述

      2. 修改cfg.training文件夹下的文件,用哪个模型就对应修改哪个文件,这里我使用的是yolov7-tiny模型,所以修改yolov7-tiny.yaml文件,VOC为20个类别所以修改为20

        在这里插入图片描述

    3. 执行命令进行训练

      python train.py --weights weights/yolov7-tiny.pt --cfg cfg/training/yolov7-tiny.yaml --data data/voc.yaml --device 0 --batch-size 16 --epoch 300
      
  7. 训练完毕得到300轮训练最好的权重,则将权重改为它,预测即可用我们自己训练的权重去预测

训练完毕得到300轮训练最好的权重,则将权重改为它,预测即可用我们自己训练的权重去预测


http://www.niftyadmin.cn/n/218771.html

相关文章

【自定义表格穿梭框】自定义封装jqgrid表格穿梭框,支持分页复选全选(附完整源码及效果图)

【写在前面】其实之前业务中也有这个方面的需求,但是总觉得自己写的有点乱,此时也就借这个机会重新封装一个公共的函数去实现这个穿梭框的效果,支持分页勾选,页面展示已选中和未选择的数据,使得系统操作更友好。 涉及知…

【软件测试】入门基础

目录 1. 软件测试的生命周期 (1)测试人员在软件开发生命周期做的事情 (2)测试人员在软件测试生命周期做的事情 (3)测试报告怎么写: 2. 如何描述一个bug 3. 如何定义bug的级别 4. bug的生命周期 …

开放原子训练营(第三季)RT-Thread Nano学习营-第一课心得

1、简介 本人学习并使用 RT-Thread 一个重要的原因是:它是一款由国内开源社区主导开发的开源实时操作系统!!! 为国产化点赞,项目中如果有要求国产化的,肯定也绕不开它。 RT-Thread支持国产化芯片&#xff…

数据库系统概论(第五版) 王珊 第四章 课后习题答案

目录 (一)数据库安全性の概念 1. 什么是数据库的安全性 2.数据库安全性和计算机系统的安全性有什么关系? 4.试述实现数据库安全性控制的常用方法和技术 5.什么是数据库中的自主存取控制方法和强制存取控制方法? (二)考试…

解决macOS IntelliJ IDEA 卡顿问题

写在前面的话1:我在撰写这篇博客时候,所用的IntelliJ IDEA版本是IntelliJ IDEA 2022.3.3 (Ultimate Edition),你需要知道可能对于不同的IntelliJ IDEA版本会有一定的差异 写在前面的话2:如果我这篇博客可以帮助到你,请…

Java模拟rank() over()函数获取分组排名的方法设计及实现

背景 考试批次班级姓名语文202302三年一班张小明130.00202302三年一班王二小128.00202302三年一班谢春花136.00202302三年二班冯世杰129.00202302三年二班马功成130.00202302三年二班魏翩翩136.00 假设我们有如上数据,现在有一个需求需要统计各学生语文单科成绩在班…

Nginx快速上手

Nginx快速上手 OVERVIEWNginx快速上手一、基本概念1.Nginx初步认识2.正向/反向代理(1)正向代理(2)反向代理二、Nginx 安装和配置1.安装2.Nginx指令3.Nginx配置三、Nginx的使用1.Web服务器(1)静态网页存储目…

【外设零基础通用教程】GPIO 上

【外设零基础通用教程】GPIO 上视频/文章结构外设:GPIO代码编写GPIO输入方法GPIO输出方法视频地址: [此内容发布后,会先给硬件群看,如果没问题才会录制视频,视频录制完成后在这里替换] 本教程适合: 硬件零…