【YOLOv5】数据增强Mosaic,把Mosaic3变成Mosaic9
Mosaic数据增强,将图片进行随机裁剪,缩放后排列接成一张图片,实现丰富数据集,增加小样本目标,提升网络的训练速度。数据增强代码在utils/dataloaders.py,找到。更改这两个地方,就可以吧Mosaic改为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
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)