【YOLOv5】数据增强Mosaic，把Mosaic3变成Mosaic9
【YOLOv5】数据增强Mosaic，把Mosaic3变成Mosaic9
云帆沧海
发布于 2024-01-17 17:23:29
发布于 2024-01-17 17:23:29
这里写自定义目录标题
- Mosaic数据增强
- 把Mosaic3变成Mosaic9
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
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2024-01-17，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录