使用albumentations库进行数据增强

2023-11-08

需要注意增强之后以字典键值的形式调用输出,且可以保证img、gt的随机增强是一一对应一致的。

import albumentations as albu

tfrd = albu.Compose([
    albu.HorizontalFlip(p=0.5),
    albu.VerticalFlip(p=0.5),
    albu.ShiftScaleRotate(shift_limit = 0,scale_limit = 0,rotate_limit=(-30,30),p=1,border_mode=cv2.BORDER_CONSTANT,value=0),
    albu.Blur(blur_limit=3, p=0.3),
])

for name in trainlist:
    img = cv2.imread(os.path.join(img_dir,name))
    gt = cv2.imread(os.path.join(gt_dir,name))
    
    for j in range(4):
        ns = tfrd(image=img,mask=gt)
        newimg,newgt= ns['image'],ns['mask']
        cv2.imwrite(os.path.join(img_dir,'t'+str(j)+'_'+name),newimg)
        cv2.imwrite(os.path.join(gt_dir,'t'+str(j)+'_'+name),newgt)
        tmp.append('t'+str(j)+'_'+name)

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用albumentations库进行数据增强 的相关文章