linux系统下训练yolov7命令

本文详细介绍YoloV7的安装配置过程、训练命令及参数调整方法,包括数据集准备、模型训练、测试验证等关键步骤,并提供训练可视化及断点续训技巧。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

关于yolov7的训练中的前期工作可以参考:

yolov7 ubuntu安装配置_安安安楠的博客-优快云博客

训练参考挺详细的:

YOLOv7训练自己的数据集(超详细)_AI小白一枚的博客-优快云博客_训练自己的数据集

这部分的重要参考:

使用yolov7训练自己的数据集; yolov7安装使用; yolov7源码解读-物联沃-IOTWORD物联网

yolo训练命令:(train运行执行命令)

python train.py --workers 16 --device 0 --batch-size 3 --data data/data.yaml  --hyp data/hyp.scratch.p5.yaml

python train.py --workers 16 --device 0 --batch-size 2 --data data/data.yaml  --hyp data/hyp.scratch.p5.yaml

设置置信度超过0.4的显示出来:(detect运行执行图像检测的命令)

python detect.py --source=inference/images/bus.jpg --weights=yolov7.pt --conf 0.4

测试摄像头:(detect运行摄像头执行的命令)

python detect.py - source 0 

python detect.py --source 0 --weights=runs/train/yolov754/weights/best.pt

python detect.py --source=scripts/Myself/images/val2007/ --weights=runs/train/exp2/weights/best.pt --conf 0.4

更改测试保存路径:(更改保存的路径)

验证模型

python test.py --data data/coco128.yaml  --weights weighs/myyolo.pt --batch-size 6

训练可视化:

conda activate yolov7           cd yolov7   执行下述操作:
tensorboard --logdir =runs 运行是成功的
tensorboard --logdir ./runs 运行是成功的
tensorboard --logdir =runs /train/exp4 运行是成功的

然后在浏览器输入 http://localhsot:6006

断点续训练

python train.py --resume runs/train/exp7/weights/last.pt

python train.py  --workers 16 --device 0 --batch-size 3 --data data/data.yaml  --weights runs/train/exp7/weights/last.pt

下载启用清华源

pip install -r requirements.txt  -i https://pypi.tuna.tsinghua.edu.cn/simple

### YOLOv5在Linux上的配置与训练 #### 1. 安装依赖库 为了顺利安装YOLOv5及其所需的各种Python包,在Linux环境下建议先更新pip并安装必要的开发工具: ```bash sudo apt-get update && sudo apt-get upgrade pip install --upgrade pip setuptools wheel ``` 对于CUDA的支持,如果计划利用GPU加速,则需确认已正确安装对应版本的CUDA和cuDNN。 #### 2. 获取YOLOv5源码 通过Git克隆官方仓库来获取最新的YOLOv5代码: ```bash git clone https://github.com/ultralytics/yolov5.git cd yolov5 ``` 此命令会下载整个项目到本地目录[^1]。 #### 3. 设置虚拟环境(推荐) 创建一个新的Python虚拟环境可以有效管理不同项目的依赖关系: ```bash python3 -m venv env source env/bin/activate pip install -r requirements.txt ``` 上述操作完成后即完成了基本软件包的准备[^2]。 #### 4. 数据集准备 按照官方文档指示准备好用于训练的数据集。通常这涉及到定义数据路径、类别名称以及划分训练集和验证集等步骤。具体做法是在`data/coco.yaml`文件中指定图像存储位置和其他参数。 #### 5. 修改配置文件 编辑位于`models/yolov5s.yaml`或其他预设模型结构下的配置文件以适应特定应用场景的需求。比如调整输入尺寸、锚框大小等超参数设置。 #### 6. 开始训练过程 一切就绪之后就可以启动实际的训练流程了。执行如下指令开始基于自定义数据集训练新的权重: ```bash python train.py --img 640 --batch 16 --epochs 50 --data data/custom_dataset.yaml --cfg models/yolov5s.yaml --weights '' ``` 这里指定了图片分辨率(`--img`)、批次大小(`--batch`)、迭代次数(`--epochs`)以及其他重要选项如所使用的数据集描述文件(`--data`)、网络架构模板(`--cfg`)还有初始化权重(`--weights`)。当不提供任何初始权重时,默认为空字符串表示随机初始化。 #### 7. 测试效果评估 完成一轮或多轮次的学习后,可以通过加载保存下来的checkpoint来进行推理测试: ```bash python detect.py --source path_to_test_image_or_video --weights runs/train/exp/weights/best.pt ``` 这条语句中的`path_to_test_image_or_video`应替换为待测样本的真实地址;而`best.pt`则是经过多次epoch后表现最好的那个checkpoint文件名。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值