def augment(components): """ Augmenting of images. :param components: components :return: updated components. """ meta = components[0] for aug in AUGMENTORS_LIST: (im, mask), params = aug.augment_return_params( AugImgMetadata(img=meta.img, mask=meta.mask, center=meta.center, scale=meta.scale)) # augment joints points = joints_to_point8(meta.all_joints) points = aug.augment_coords(points, params) # after flipping horizontaly the left side joints and right side joints are also # flipped so we need to recover their orginal orientation. if isinstance(aug, FlipAug): points = aug.recover_left_right(points, params) # augment center position new_center = aug.augment_coords(meta.center, params) meta.all_joints = point8_to_joints(points) meta.center = new_center meta.img = im meta.mask = mask return components
def augment(components): """ Augmenting of images. :param components: components :return: updated components. """ meta = components[0] aug_center = meta.center.copy() aug_joints = joints_to_point8(meta.all_joints) for aug in AUGMENTORS_LIST: (im, mask), params = aug.augment_return_params( AugImgMetadata(img=meta.img, mask=meta.mask, center=aug_center, scale=meta.scale)) # augment joints if AUGMENTORS_LIST.index(aug) == 2: meta.crop_x = aug.crop_x meta.crop_y = aug.crop_y meta.crop_x_max = aug.crop_x_max meta.crop_y_max = aug.crop_y_max params = (params[0] - int(aug.crop_x_max / 2 - aug.crop_x / 2), params[1] - int(aug.crop_y_max / 2 - aug.crop_y / 2)) aug_joints = aug.augment_coords(aug_joints, params) # after flipping horizontaly the left side joints and right side joints are also # flipped so we need to recover their orginal orientation. #if isinstance(aug, FlipAug): # aug_joints = aug.recover_left_right(aug_joints, params) # augment center position aug_center = aug.augment_coords(aug_center, params) meta.img = im meta.mask = mask meta.aug_joints = point8_to_joints(aug_joints) meta.aug_center = aug_center return components
def augment(components): #数据增强是随机的吗?是每个样本都处理还是随机处理,增强方法右四种,用的是哪一个?? """ Augmenting of images. :param components: components :return: updated components. """ meta = components[0] aug_center = meta.center.copy() aug_joints = joints_to_point8(meta.all_joints) for aug in AUGMENTORS_LIST: (im, mask), params = aug.augment_return_params( AugImgMetadata(img=meta.img, mask=meta.mask, center=aug_center, scale=meta.scale)) # augment joints aug_joints = aug.augment_coords(aug_joints, params) # after flipping horizontaly the left side joints and right side joints are also # flipped so we need to recover their orginal orientation. if isinstance(aug, FlipAug): aug_joints = aug.recover_left_right(aug_joints, params) # augment center position aug_center = aug.augment_coords(aug_center, params) meta.img = im meta.mask = mask meta.aug_joints = point8_to_joints(aug_joints) meta.aug_center = aug_center return components