【YOLOv5】数据增强Mosaic,把Mosaic3变成Mosaic9
yolov5
yolov5 - Ultralytics YOLOv8的前身,是一个用于目标检测、图像分割和图像分类任务的先进模型。
项目地址:https://gitcode.com/gh_mirrors/yo/yolov5
免费下载资源
·
这里写自定义目录标题
Mosaic数据增强
Mosaic数据增强,将图片进行随机裁剪,缩放后排列接成一张图片,实现丰富数据集,增加小样本目标,提升网络的训练速度。
把Mosaic3变成Mosaic9
数据增强代码在utils/dataloaders.py,找到
mosaic = self.mosaic and random.random() < hyp['mosaic']
if mosaic:
# Load mosaic
img, labels = load_mosaic(self, index) # use load_mosaic4
shapes = None
# MixUp augmentation
if random.random() < hyp['mixup']:
img, labels = mixup(img, labels, *load_mosaic(self, random.randint(0, self.n - 1)))
把其中的:
img, labels = load_mosaic(self, index)
改为:
img, labels = load_mosaic9(self, index)
然后:
img, labels = mixup(img, labels, *load_mosaic(self, random.randint(0, self.n - 1)))
改为:
img, labels = mixup(img, labels, *load_mosaic9(self, random.randint(0, self.n - 1)))
更改这两个地方,就可以吧Mosaic改为Mosaic9
GitHub 加速计划 / yo / yolov5
49.4 K
16.03 K
下载
yolov5 - Ultralytics YOLOv8的前身,是一个用于目标检测、图像分割和图像分类任务的先进模型。
最近提交(Master分支:2 个月前 )
79b7336f
* Update Integrations table
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.zh-CN.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> 20 天前
94a62456
* fix: quad training
* fix: quad training in segmentation 22 天前
更多推荐
已为社区贡献2条内容
所有评论(0)