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
