def set_transform(self):
if self.augments:
color_jitter = OneOf(transforms=[
Identity(),
HSV(p=1,
hgain=self.aug_cfg['hsv_h'],
sgain=self.aug_cfg['hsv_s'],
vgain=self.aug_cfg['hsv_v']),
RandNoise()
])
basic_transform = Compose(transforms=[
color_jitter,
ScaleMinMax(min_thresh=self.min_thresh,
max_thresh=self.max_thresh),
])
mosaic_transform = Mosaic(self.img_paths,
self.labels,
color_gitter=color_jitter,
target_size=self.max_thresh,
pad_val=self.aug_cfg['pad_val'],
rand_center=True)
self.transform = Compose(transforms=[
OneOf(transforms=[(0.5,
basic_transform), (0.5, mosaic_transform)]),
LRFlip()
])
else:
self.transform = ScaleMinMax(min_thresh=self.min_thresh,
max_thresh=self.max_thresh)
def set_transform(self):
if self.augments:
color_jitter = OneOf(transforms=[
Identity(),
HSV(p=1,
hgain=self.aug_cfg['hsv_h'],
sgain=self.aug_cfg['hsv_s'],
vgain=self.aug_cfg['hsv_v']),
RandNoise()
])
mosaic = Mosaic(self.img_paths,
self.labels,
color_gitter=color_jitter,
target_size=self.img_size,
pad_val=self.aug_cfg['pad_val'],
rand_center=True)
mix_up = MixUp(self.img_paths,
self.labels,
color_gitter=color_jitter,
target_size=self.img_size,
pad_val=self.aug_cfg['pad_val'],
beta=self.aug_cfg['beta'])
perspective_transform = RandPerspective(
target_size=(self.img_size, self.img_size),
scale=self.aug_cfg['scale'],
degree=self.aug_cfg['degree'],
translate=self.aug_cfg['translate'],
shear=self.aug_cfg['shear'],
pad_val=self.aug_cfg['pad_val'])
basic_transform = Compose(transforms=[
color_jitter,
RandCutOut(),
ScalePadding(target_size=self.img_size,
padding_val=self.aug_cfg['pad_val']),
perspective_transform,
])
aug_mosaic = Mosaic(self.img_paths,
self.labels,
color_gitter=mix_up,
target_size=self.img_size,
pad_val=self.aug_cfg['pad_val'],
rand_center=True)
aug_mixup = MixUp(self.img_paths,
self.labels,
color_gitter=mosaic,
target_size=self.img_size,
pad_val=self.aug_cfg['pad_val'],
beta=self.aug_cfg['beta'])
self.transform = Compose(transforms=[
OneOf(transforms=[(0., basic_transform), (
1., mosaic), (0., mix_up), (0., aug_mosaic), (0.,
aug_mixup)]),
LRFlip()
])
else:
self.transform = ScalePadding(target_size=(self.img_size,
self.img_size),
padding_val=self.aug_cfg['pad_val'])
def set_transform(self):
if self.augments:
self.transform = Compose(transforms=[
OneOf(transforms=[
(0.6, Compose(transforms=[
OneOf(transforms=[Identity(),
HSV(p=1,
hgain=self.aug_cfg['hsv_h'],
sgain=self.aug_cfg['hsv_s'],
vgain=self.aug_cfg['hsv_v']),
RandNoise()
]),
RandCutOut(),
ScalePadding(target_size=self.img_size, padding_val=self.aug_cfg['pad_val']),
RandPerspective(target_size=(self.img_size, self.img_size),
scale=self.aug_cfg['scale'],
degree=self.aug_cfg['degree'],
translate=self.aug_cfg['translate'],
shear=self.aug_cfg['shear'],
pad_val=self.aug_cfg['pad_val'])])),
(0.4, Mosaic(self.img_paths,
self.labels,
color_gitter=OneOf(transforms=[Identity(),
HSV(p=1,
hgain=self.aug_cfg['hsv_h'],
sgain=self.aug_cfg['hsv_s'],
vgain=self.aug_cfg['hsv_v']),
RandNoise()]),
target_size=self.img_size,
pad_val=self.aug_cfg['pad_val']))
]),
LRFlip()])
else:
self.transform = ScalePadding(target_size=(self.img_size, self.img_size),
padding_val=self.aug_cfg['pad_val'])