yolov5进行识别安全帽

下载yolov5

地址是:https://github.com/ultralytics/yolov5

使用按照教程来进行就行

这是教程

这里简单说一下:

  • 下载需要的命令: pip install -r requirements.txt

  • **注意环境:**克隆 repo,并要求在 Python>=3.8.0 环境中安装 requirements.txt ,且要求 PyTorch>=1.8 。 .org/get-started/locally/) 。

  • 想用GPU,记得下载CUDA

  • 运行指令: python train.py --data coco.yaml --epochs 300 --weights ‘’ --cfg yolov5n.yaml --batch-size 128

  • 执行指令(举例): python detect.py --weights yolov5s.pt --source 0

注意事项(有必要看看)

这里用coco.yaml和yolov5n.yaml进行举例子

  • 首先想要训练自己的数据集,记得把路径改好
    coco

  • 类别数与yolov5n.yaml中要一致
    coco
    yolov5

  • 使用coco_hat.yaml也是可以的,可以仿照它的路径进行修改
    coco_hat

  • 说明一下自己的数据集
    格式
    images\train2017中是jpg的格式,其他格式可能会报错
    labels\train2017中的数据是txt文件
    举例
    每行的意思是:类别的数据(在coco.yaml中对应),中心横坐标,纵坐标,宽,高

  • 训练完成后会在终端输出保存的路径,其中weights中有训练模型文件(.pt文件),要想执行,进行使用,记得将yolov5n.pt文件换成weights中文件

  • 当–batch-size过大时,会超出电脑承受力,会报错的,可以在网上租一个云计算进行训练

效果

效果

总结

这是好久之前做的了,今天整理博客才发现,其实并没有写完,但是就这样也可以了,其实很简单的。

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-22 08:52:04       101 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-22 08:52:04       109 阅读
  3. 在Django里面运行非项目文件

    2024-07-22 08:52:04       87 阅读
  4. Python语言-面向对象

    2024-07-22 08:52:04       96 阅读

热门阅读

  1. 华为eNSP模拟器安装

    2024-07-22 08:52:04       22 阅读
  2. HTTP协议的演进:从HTTP/1.0到HTTP/2.0

    2024-07-22 08:52:04       19 阅读
  3. 在Ubuntu 14.04上安装和使用Docker Compose的方法

    2024-07-22 08:52:04       24 阅读
  4. 【自动化机器学习AutoML】AutoML工具和平台的使用

    2024-07-22 08:52:04       21 阅读
  5. 【数据挖掘基础】数据挖掘技术概述和基本算法

    2024-07-22 08:52:04       26 阅读
  6. 常用传感器误差补偿方法介绍

    2024-07-22 08:52:04       24 阅读
  7. ARM/Linux嵌入式面经(十七):美团校招面经

    2024-07-22 08:52:04       23 阅读
  8. 深度学习简介(框架)

    2024-07-22 08:52:04       22 阅读
  9. ChatGPT的工作记忆容量:一项实证研究

    2024-07-22 08:52:04       17 阅读
  10. AI学习指南机器学习篇-SOM的拓扑结构与参数调优

    2024-07-22 08:52:04       24 阅读
  11. 如何调整图像的窗宽窗位

    2024-07-22 08:52:04       25 阅读
  12. linux字符设备驱动+fops应用测试程序

    2024-07-22 08:52:04       20 阅读
  13. opencv—常用函数学习_“干货“_14

    2024-07-22 08:52:04       25 阅读
  14. 网络安全防线:黑龙江等级保护测评标准详解

    2024-07-22 08:52:04       23 阅读
  15. thinkphp8结合layui2.9 图片上传验证

    2024-07-22 08:52:04       24 阅读
  16. grub之loongarch架构调试

    2024-07-22 08:52:04       17 阅读