
图像增强
有点方
这个作者很懒,什么都没留下…
展开
-
数据增强Mosaic
丰富检测物体的背景,且在BN计算的时候一下子会计算四张图片的数据,使得mini-batch大小不需要很大,那么一个GPU就可以达到比较好的效果。实现思路1、每次读取四张图片。2、分别对四张图片进行翻转、缩放、色域变化等,并且按照四个方向位置摆好。3、进行图片的组合和框的组合主要代码: def load_mosaic(self, index): # loads images in a mosaic s = self.img_size // 2..原创 2020-08-05 14:56:21 · 1586 阅读 · 0 评论 -
CutMix
Mixup:将随机的两张样本按比例混合,分类的结果按比例分配; Cutout:随机的将样本中的部分区域cut掉,并且填充0像素值,分类的结果不变; CutMix随机生成一个裁剪框Box,裁剪掉A图的相应位置,然后用B图片相应位置的ROI放到A图中被裁剪的区域形成新的样本,计算损失时同样采用加权求和的方式进行求解主要code(Iteration过程中的实现方法):lam = np.random.beta(args.beta, args.beta)rand_index = torch.ran...原创 2020-08-05 14:37:28 · 649 阅读 · 0 评论