【YOLOv5】数据增强Mosaic，把Mosaic3变成Mosaic9

这里写自定义目录标题

• Mosaic数据增强
  • 把Mosaic3变成Mosaic9

Mosaic数据增强

Mosaic数据增强,将图片进行随机裁剪，缩放后排列接成一张图片，实现丰富数据集，增加小样本目标，提升网络的训练速度。

把Mosaic3变成Mosaic9

数据增强代码在utils/dataloaders.py,找到

代码语言：javascript复制

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)))

把其中的：

代码语言：javascript复制

img, labels = load_mosaic(self, index)

改为：

代码语言：javascript复制

img, labels = load_mosaic9(self, index)   

然后：

代码语言：javascript复制

 img, labels = mixup(img, labels, *load_mosaic(self, random.randint(0, self.n - 1)))

改为：

代码语言：javascript复制

 img, labels = mixup(img, labels, *load_mosaic9(self, random.randint(0, self.n - 1)))

更改这两个地方，就可以吧Mosaic改为Mosaic9

数据 网络 load random self

0 人点赞