Ubuntu YOLOv5自制训练集训练

Ubuntu系统 用自制的数据集训练YOLOv5网络

1. 本人使用的YOLOv5网络 https://github.com/Thinkin99/yolov5_d435i_detection

2. 训练用的是官方的网络 https://github.com/ultralytics/yolov5,然后把训练好的模型导入自己所用的网络中使用。

3. 训练集的制作、分割和转换主要参考 https://blog.csdn.net/qq_40927867/article/details/115768888, 虽然博主是说的WIN系统下,但Ubuntu一样可以使用。

4. 需要注意的是,Github上官方最新版的YOLOv5训练完的模型可能不适合所有版本的网络,我在网盘上存了一个自己可以使用的版本 链接:https://pan.baidu.com/s/1AN–Tu_xowqkMDiX7YXlNg?pwd=50y7 提取码:50y7

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
YOLO系列是基于深度学习的端到端实时目标检测方法。 PyTorch版的YOLOv5轻量而高性能,更加灵活和易用,当前非常流行。 本课程将手把手地教大家使用labelImg标注和使用YOLOv5训练自己的数据集。课程实战分为两个项目:单目标检测(足球目标检测)和多目标检测(足球和梅西同时检测)。  本课程的YOLOv5使用ultralytics/yolov5,在Windows和Ubuntu系统上分别做项目演示。包括:安装YOLOv5、标注自己的数据集、准备自己的数据集(自动划分训练集和验证集)、修改配置文件、使用wandb训练可视化工具、训练自己的数据集、测试训练出的网络模型和性能统计。 除本课程《YOLOv5实战训练自己的数据集(Windows和Ubuntu演示)》外,本人推出了有关YOLOv5目标检测的系列课程。请持续关注该系列的其它视频课程,包括:《YOLOv5(PyTorch)目标检测:原理与源码解析》课程链接:https://edu.csdn.net/course/detail/31428《YOLOv5目标检测实战:Flask Web部署》课程链接:https://edu.csdn.net/course/detail/31087《YOLOv5(PyTorch)目标检测实战:TensorRT加速部署》课程链接:https://edu.csdn.net/course/detail/32303《YOLOv5目标检测实战:Jetson Nano部署》课程链接:https://edu.csdn.net/course/detail/32451《YOLOv5+DeepSORT多目标跟踪与计数精讲》课程链接:https://edu.csdn.net/course/detail/32669《YOLOv5实战口罩佩戴检测》课程链接:https://edu.csdn.net/course/detail/32744《YOLOv5实战中国交通标志识别》课程链接:https://edu.csdn.net/course/detail/35209 《YOLOv5实战垃圾分类目标检测》课程链接:https://edu.csdn.net/course/detail/35284  
要在Ubuntu上使用YOLOv5训练自己的数据集,可以按照以下步骤进行操作: 1. 安装必要的依赖项:在Ubuntu上安装Python、PyTorch、OpenCV和其他相关的软件包。可以使用以下命令在终端中安装它们: ```shell pip install torch torchvision opencv-python ``` 2. 下载YOLOv5代码:从YOLOv5的官方GitHub仓库中克隆代码到本地。使用以下命令在终端中执行: ```shell git clone https://github.com/ultralytics/yolov5.git ``` 3. 准备数据集:将自己的数据集标注成YOLO格式,并将图像和标签文件放入`yolov5/data`文件夹中。 4. 配置数据集:根据自己的数据集,修改`yolov5/data`文件夹中的`yaml`配置文件。主要包括`train`和`val`数据集的路径、类别数量等信息。 5. 训练模型:使用以下命令开始训练模型: ```shell cd yolov5 python train.py --img <image_size> --batch <batch_size> --epochs <num_epochs> --data data/<your_yaml_file.yaml> --cfg models/yolov5s.yaml --weights '' --name <experiment_name> ``` 其中, `<image_size>` 是输入图像的尺寸,`<batch_size>` 是每批次处理的图像数量, `<num_epochs>` 是训练的迭代次数, `<your_yaml_file.yaml>` 是你修改后的`yaml`配置文件名,`<experiment_name>` 是保存实验结果的文件夹名。 6. 测试模型:使用以下命令测试已训练的模型,在终端中执行: ```shell python detect.py --weights runs/train/<experiment_name>/weights/best.pt --img <image_path> --conf 0.4 ``` 其中, `<experiment_name>` 是你之前训练模型时使用的实验文件夹名,`<image_path>` 是要进行目标检测的图像路径, `--conf 0.4` 是置信度阈值。 以上是在Ubuntu上使用YOLOv5训练自己的数据集的一般步骤。根据实际情况,可能需要额外的配置和调整。请参考YOLOv5的官方文档和GitHub仓库中的说明进行更详细的操作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值