MOSAIC数据增强

MOSAIC代码

def mosaicDataAugment(img):
    images = []
    images.append(img)
    images.append(img)
    images.append(img)
    images.append(img)
    
    xc, yc = np.random.randint(320, 960, 2)
    print(xc, yc)
    s = 640
    for i, img in enumerate(images):
        h, w, c = img.shape
        
        if i == 0:
            img4 = np.full((s*2, s*2, c), 114, dtype=np.uint8)
            x1a, y1a, x2a, y2a = max(xc - w, 0), max(yc - h, 0), xc, yc
            x1b, y1b, x2b, y2b = w - (x2a - x1a), h - (y2a - y1a), w, h
        elif i == 1:
            x1a, y1a, x2a, y2a = xc, max(yc - h, 0), min(xc + w, s * 2), yc
            x1b, y1b, x2b, y2b = 0, h - (y2a - y1a), x2a - x1a, h
        elif i == 2:
            x1a, y1a, x2a, y2a = max(xc - w, 0), yc, xc, min(yc + h, s * 2)
            x1b, y1b, x2b, y2b = w - (x2a - x1a), 0, w, y2a - y1a
        elif i == 3:
            x1a, y1a, x2a, y2a = xc, yc, min(xc + w, s * 2), min(yc + h, s * 2)
            x1b, y1b, x2b, y2b = 0, 0, x2a - x1a, y2a - y1a
    
        img4[y1a:y2a, x1a:x2a] = img[y1b:y2b, x1b:x2b]
    
    cv2.imshow("aa", img4)
    cv2.waitKey()
   
if __name__ == "__main__":
    img = cv2.imread("arg.jpg")
    mosaicDataAugment(img)

图片