当前位置: 首页 > news >正文

目标检测YOLO系列-YOLOV7运行步骤(推理、训练全过程)

  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.lryc.cn/news/60471.html

相关文章:

  • Spring Boot + Spring Security基础入门教程
  • MySQL数据库,表的增删改查详细讲解
  • SpringCloud-Gateway实现网关
  • Redis 如何配置读写分离架构(主从复制)?
  • 代码随想录二刷day05 | 哈希表之242.有效的字母异位词 349. 两个数组的交集 202. 快乐数 1. 两数之和
  • 2023年4月广东省计算机软考中/高级备考班招生简章
  • 在Github中77k星的王炸AutoGPT,会独立思考,直接释放双手
  • FVM链的Themis Pro,5日ido超百万美元
  • OpenCV实战——尺度不变特征检测器
  • 如何快速建立一个podman环境
  • 计算机视觉:人工智能领域当下火热的计算机视觉技术综述
  • EMC 专用名词大全~骚扰波形
  • 14:24面试,14:32就出来了 ,问的实在是太...
  • 高频算法题
  • AI工程师眼中的未来 | 年轻人如何求职选方向
  • 能自动翻译的软件-最精准的翻译软件
  • 7.1 大学排行榜分析(project)
  • TensorFlow 2.0 的新增功能:第三、四部分
  • 第1章 如何听起来像数据科学家
  • 哈希表题目:在系统中查找重复文件
  • 机器人感知与控制关键技术及其智能制造应用
  • 精通线程池,看这一篇就够了
  • 解决图片、视频地址加密问题
  • GPT引领学习之旅:一篇让程序员轻松掌握Elasticsearch的攻略
  • 23种设计模式-仲裁者模式(Android应用场景介绍)
  • 【数据统计】— 极大似然估计 MLE、最大后验估计 MAP、贝叶斯估计
  • Zookeeper学习笔记
  • go语言切片做函数参数传递+append()函数扩容
  • 2023.04.16 学习周报
  • 【面试】如何设计SaaS产品的数据权限?