这里写自定义目录标题

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

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐