本文包含了有关yolov5目标检测的基本流程,包括模型训练与模型部署,旨在帮助小伙伴们建立系统的认知
YOLO是 "You only look once "的首字母缩写,是一个开源软件工具,它具有实时检测特定图像中物体的高效能力。YOLO算法使用卷积神经网络(CNN)模型来检测图像中的物体。该算法只需要通过给定的神经网络进行一次前向传播就能检测到图像中的所有物体。这使YOLO算法在速度上比其他算法更有优势,使其成为迄今为止最著名的检测算法之一。
Win 10
Python 3.8
PyTorch 1.12.0
Pycharm professional 2022
pip install -r requirements.txt
(自动安装其他环境)
pip install wanbd
pip install tensorboard
pip install pyyaml
pip install ipython
pip install openvino
pip install opencv-python-headless
文件目录
注:models是权重配置,没写全
简洁运行:
python train.py
带参数(只列出部分重要参数):
大部分可以直接在train.py文件中parse_opt函数中直接修改。
下面这些不是的,必须在配置文件中修改
--hyp
选项中,默认为data/hyps/hyp.scratch.yaml文件;--data
选项中,默认为data/coco128.yaml文件;--data
选项中,默认为data/coco128.yaml文件;文件快速定位
yolov5 网络结构详细介绍
简洁运行:
python val.py
带参数:
yolov5的评价指标详细介绍
简洁运行:
python detect.py
带参数:
到这里差不多啦,大家可以看一下专栏中的其他yolov5作品,里面有更加详细的区域介绍。
欢迎点赞关注啦