def load_augmentation_settings(handler_save_path: str):
if not file_exists(handler_save_path):
handler = AugHandler(
[
aug.Affine(
scale = {"x": tuple([0.8, 1]), "y":tuple([0.8, 1])},
translate_percent= {"x": tuple([0.1, 0.11]), "y":tuple([0.1, 0.11])},
rotate= [-180, 180],
order= [0, 0],
cval= [0, 0],
shear= [0,0],
frequency= 0.6).change_rotate_to_right_angle(),
aug.GaussianBlur(sigma=(0, 1.5),frequency=0.5),
#aug.Crop(percent=[0, 0.1], frequency= 0.5),
#aug.Flipud(p=0.5),
aug.Sharpen(alpha=[0,0.5], lightness=[0.8,1], frequency= 0.5),
aug.Emboss(alpha=[0,0.5], strength=[0.8,1], frequency=0.5),
aug.AdditiveGaussianNoise(loc= 0, scale=[0,24.75], per_channel=1,frequency=0.8),
aug.Add(value=[-20,20], per_channel=True, frequency= 0.5),
aug.LinearContrast(alpha=[0.6,1.4], per_channel=True,frequency= 0.5),
aug.Dropout(p=[0,0.01], per_channel=True),
aug.AverageBlur(k=[1,2], frequency = 0.5),
aug.MotionBlur(k=[3,5], angle=[0,360], frequency = 0.5),
aug.Fliplr(p=0.5, frequency=0.7)
]
)
handler.save_to_path(save_path=handler_save_path, overwrite=True)
else:
handler = AugHandler.load_from_path(handler_save_path)
return handler
def load_augmentation_settings(handler_save_path: str):
if not file_exists(handler_save_path):
handler = AugHandler(
[
aug.Crop(percent=[0.2, 0.5]),
aug.Affine(scale = {"x": tuple([0.8, 1.2]), "y":tuple([0.8, 1.2])}, translate_percent= {"x": tuple([0, 0]), "y":tuple([0, 0])}, rotate= [0, 0], order= [0, 0], cval= [0, 0], shear= [0,0], fit_output=True)
]
)
handler.save_to_path(save_path=handler_save_path, overwrite=True)
else:
handler = AugHandler.load_from_path(handler_save_path)
return handler
handler = AugHandler(aug_modes=[
# aug.Affine(
# scale = {"x": tuple([0.8, 1]), "y":tuple([0.8, 1])},
# translate_percent= {"x": tuple([0.1, 0.11]), "y":tuple([0.1, 0.11])},
# rotate= [-180, 180],
# order= [0, 0],
# cval= [0, 0],
# shear= [0,0],
# frequency= 0.6).change_rotate_to_right_angle(),
#aug.GaussianBlur(sigma=(0, 1.5),frequency=0.5),
#aug.Crop(percent=[0, 0.05], frequency= 0.5),
#aug.Flipud(p=0.5),
#aug.Superpixels(),
#aug.Sharpen(alpha=[0,0.5], lightness=[0.8,1], frequency= 0.5),
#aug.Emboss(alpha=[0,0.5], strength=[0.8,1], frequency=0.5),
#aug.AdditiveGaussianNoise(loc= 0, scale=[0,24.75], per_channel=1,frequency=0.8),
#aug.Add(value=[-20,20], per_channel=True, frequency= 0.5),
#aug.LinearContrast(alpha=[0.6,1.4], per_channel=True,frequency= 0.5),
#aug.Dropout(p=[0,0.1], per_channel=True),
#aug.AverageBlur(k=[1,2], frequency = 0.5),
#aug.MotionBlur(k=[3,5], angle=[0,360], frequency = 0.5),
# aug.Invert(p=0.5, per_channel=False,frequency=1),
#aug.Grayscale(alpha=0.2),
# aug.Multiply(mul=[0.8,1.2], per_channel=True),
# aug.ElasticTransformation(alpha=[0,40], sigma=[4,6]),
# aug.PiecewiseAffine(scale=[0.0,0.05]),
#aug.ContrastNormalization(alpha=[0.7,1], per_channel=True),
# iaa.BilateralBlur(d=(7, 7), sigma_color=(10, 250), sigma_space=(10, 250))
#aug.Fliplr(p=0.5, frequency=0.7)
# aug.BilateralBlur(d=[1,9]),
#aug.EdgeDetect(alpha=[0,0.1]),
# aug.DirectedEdgeDetect(alpha=[0,0.5], direction=[0,1]),
# aug.CoarseDropout(p=[0,0.04], size_percent=0.05),
# aug.Resize()
])
def __init__(self, cfg, aug_settings_file_path, is_train=True):
super().__init__(cfg, is_train=is_train)
self.aug_handler = AugHandler.load_from_path(aug_settings_file_path)
dataset = COCO_Dataset.load_from_path(
json_path='/home/pasonatech/combined_cocooutput/HSR-coco.json',
img_dir='/home/pasonatech/combined_cocooutput'
#json_path='/home/pasonatech/aug_real_combine/aug_sim_com_garbage/HSR-coco.json',
#img_dir='/home/pasonatech/aug_real_combine/aug_sim_com_garbage'
)
resize_save_path = 'test_resize.json'
handler_save_path = 'test_handler.json'
if not file_exists(handler_save_path):
handler = AugHandler(
[
aug.Crop(percent=[0.2, 0.5]),
aug.Flipud(p=0.5),
aug.Superpixels()
# aug.Sharpen(alpha=[-1,0.1], lightness=[0,3])
]
)
handler.save_to_path(save_path=handler_save_path, overwrite=True)
logger.info(f'Created new AugHandler save.')
else:
handler = AugHandler.load_from_path(handler_save_path)
logger.info(f'Loaded AugHandler from save.')
i=0
img_buffer = []
for coco_image in dataset.images:
img = cv2.imread(coco_image.coco_url)
img_buffer.append(img)
keypoints = []
'/Users/darwinharianto/Desktop/hayashida/Unreal/18_03_2020_18_03_10_coco-data'
)
resize_save_path = 'test_resize.json'
handler_save_path = 'test_handler.json'
if not file_exists(resize_save_path):
resize = aug.Resize(width=500, height=500)
resize.save_to_path(save_path=resize_save_path, overwrite=True)
logger.info(f'Created new Resize save.')
else:
resize = aug.Resize.load_from_path(resize_save_path)
logger.info(f'Loaded Resize from save.')
if not file_exists(handler_save_path):
handler = AugHandler([
aug.Crop(percent=[0.2, 0.5]),
aug.Flipud(p=0.5),
aug.Superpixels()
# aug.Sharpen(alpha=[-1,0.1], lightness=[0,3])
])
handler.save_to_path(save_path=handler_save_path, overwrite=True)
logger.info(f'Created new AugHandler save.')
else:
handler = AugHandler.load_from_path(handler_save_path)
handler = AugHandler(aug_modes=[
# aug.Affine(scale = {"x": tuple([0.8, 1.2]), "y":tuple([0.8, 1.2])}, translate_percent= {"x": tuple([0.1, 0.11]), "y":tuple([0.1, 0.11])}, rotate= [-180, 180], order= [0, 0], cval= [0, 0], shear= [0,0]),
# aug.Crop(percent=[0.2, 0.5]),
# aug.Flipud(p=0.5),
# aug.Superpixels(),
# aug.Sharpen(alpha=[0,0.1], lightness=[0.8,1]),
# aug.Emboss(alpha=[0,0.1], strength=[0.8,1]),
# aug.AdditiveGaussianNoise(),
# aug.Invert(p=1, per_channel=False),