YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）-Toy模板网

这篇具有很好参考价值的文章主要介绍了YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

前言

由于本人水平有限，难免出现错漏，敬请批评改正。

更多精彩内容，可点击进入YOLO系列专栏或我的个人主页查看

YOLOv5：添加SE、CBAM、CoordAtt、ECA注意力机制

YOLOv5：yolov5s.yaml配置文件解读、增加小目标检测层

YOLOv5：IoU、GIoU、DIoU、CIoU、EIoU

YOLOv7训练自己的数据集（口罩检测）

YOLOv8训练自己的数据集（足球检测）

玩转Jetson Nano（五）：TensorRT加速YOLOv5目标检测

前提条件

熟悉Python

相关介绍

Python是一种跨平台的计算机程序设计语言。是一个高层次的结合了解释性、编译性、互动性和面向对象的脚本语言。最初被设计用于编写自动化脚本(shell)，随着版本的不断更新和语言新功能的添加，越多被用于独立的、大型项目的开发。

PyTorch 是一个深度学习框架，封装好了很多网络和深度学习相关的工具方便我们调用，而不用我们一个个去单独写了。它分为 CPU 和 GPU 版本，其他框架还有 TensorFlow、Caffe 等。PyTorch 是由 Facebook 人工智能研究院（FAIR）基于 Torch 推出的，它是一个基于 Python 的可续计算包，提供两个高级功能：1、具有强大的 GPU 加速的张量计算（如 NumPy）；2、构建深度神经网络时的自动微分机制。

YOLOv5是一种单阶段目标检测算法，该算法在YOLOv4的基础上添加了一些新的改进思路，使其速度与精度都得到了极大的性能提升。它是一个在COCO数据集上预训练的物体检测架构和模型系列，代表了Ultralytics对未来视觉AI方法的开源研究，其中包含了经过数千小时的研究和开发而形成的经验教训和最佳实践。

实例分割是指将图片中属于物体类别的像素识别出来并作分类。1 实例分割是视觉经典四个任务中相对最难的一个，它既具备语义分割（Semantic Segmentation）的特点，需要做到像素层面上的分类，也具备目标检测（Object Detection）的一部分特点，即需要定位出不同实例，即使它们是同一种类。

本文数据集免费获取链接：https://download.csdn.net/download/FriendshipTang/88118028

同时，也可以在本文开头获取，如下图所示。

使用YOLOv5-7.0版本训练自己的实例分割模型

YOLOv5项目官方源地址

https://github.com/ultralytics/yolov5.git

下载yolov5-7.0版源码

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

解压

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

目录结构

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

准备实例分割数据集

在./data目录下，新建myseg.yaml配置文件

内容如下：

# YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
# COCO128-seg dataset https://www.kaggle.com/ultralytics/coco128 (first 128 images from COCO train2017) by Ultralytics
# Example usage: python train.py --data coco128.yaml
# parent
# ├── yolov5
# └── datasets
#     └── coco128-seg  ← downloads here (7 MB)


# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: ../datasets/myseg  # dataset root dir
train: ../myseg/trainset/images  # train images (relative to 'path') 128 images
val: ../myseg/valset/images  # val images (relative to 'path') 128 images
test: ../myseg/testset/images  # test images (optional)

# Classes
names:
  0: background
  1: car
  2: traffic_sign
  3: lane_lines
  4: person
  5: motorcyclist
  6: cyclist

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

训练分割模型

python segment/train.py --data data/myseg.yaml --weights yolov5s-seg.pt --img 640 --batch-size 16

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

注：如果报显存溢出错误，可将batch-size调小。

解决‘ImportError: Failed to initialize: Bad git executable.’

Traceback (most recent call last):
  File "segment/train.py", line 66, in <module>
    GIT_INFO = check_git_info()
  File "D:\anaconda3\envs\test2\lib\contextlib.py", line 75, in inner
    return func(*args, **kwds)
  File "E:\mytest\yolov5-master\utils\general.py", line 360, in check_git_info
    import git
  File "D:\anaconda3\envs\test2\lib\site-packages\git\__init__.py", line 91, in <module>
    raise ImportError("Failed to initialize: {0}".format(exc)) from exc
ImportError: Failed to initialize: Bad git executable.
The git executable must be specified in one of the following ways:
    - be included in your $PATH
    - be set via $GIT_PYTHON_GIT_EXECUTABLE
    - explicitly set via git.refresh()

All git commands will error until this is rectified.

This initial warning can be silenced or aggravated in the future by setting the
$GIT_PYTHON_REFRESH environment variable. Use one of the following values:
    - quiet|q|silence|s|none|n|0: for no warning or exception
    - warn|w|warning|1: for a printed warning
    - error|e|raise|r|2: for a raised exception

Example:
    export GIT_PYTHON_REFRESH=quiet

解决方法

在train.py代码中，import os 后面添加了一行

os.environ["GIT_PYTHON_REFRESH"] = "quiet"

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

验证分割模型

python segment/val.py --weights runs\train-seg\exp\weights\best.pt --data data/myseg.yaml --img 640

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

BoxPR_curve

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

MaskPR_curve

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

测试分割模型

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

python segment/predict.py --weights runs/train-seg/exp/weights/best.pt --source test.jpg --img 640

YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）,YOLO系列,OpenCV-Python小应用,YOLO,深度学习

参考

[1] https://github.com/ultralytics/yolov5文章来源地址https://www.toymoban.com/news/detail-615539.html

由于本人水平有限，难免出现错漏，敬请批评改正。

更多精彩内容，可点击进入YOLO系列专栏或我的个人主页查看

YOLOv5：添加SE、CBAM、CoordAtt、ECA注意力机制

YOLOv5：yolov5s.yaml配置文件解读、增加小目标检测层

YOLOv5：IoU、GIoU、DIoU、CIoU、EIoU

YOLOv7训练自己的数据集（口罩检测）

YOLOv8训练自己的数据集（足球检测）

玩转Jetson Nano（五）：TensorRT加速YOLOv5目标检测

到了这里，关于YOLOv5：使用7.0版本训练自己的实例分割模型（车辆、行人、路标、车道线等实例分割）的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！