-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathaugmentation.py
More file actions
63 lines (57 loc) · 3.94 KB
/
augmentation.py
File metadata and controls
63 lines (57 loc) · 3.94 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import numpy as np
import albumentations as A
import cv2
def flip_up_down(image, ground_truth):
# print('Flipping Data up and down')
# print(f'recive : image : {image.shape} {image.dtype} :: ground_truth { ground_truth.shape} { ground_truth.dtype} ')
transform = A.Compose([A.VerticalFlip(p=1.0)])
transformed_image = transform(image=image)['image']
transformed_gt = transform(image=ground_truth)['image']
# print(f'data Typye check: \n transformed_image -> shape : {transformed_image.shape} dataytpe: {transformed_image.dtype}:: \n transformed_gt :: shape {transformed_gt.shape} dtype {transformed_gt.dtype} ')
# print()
return transformed_image, transformed_gt,
def flip_left_right(image, ground_truth):
# print('Shifting data left to right on x axis')
# print(f'recive : image : {image.shape} {image.dtype} :: ground_truth { ground_truth.shape} { ground_truth.dtype} ')
transform = A.Compose([A.HorizontalFlip(p=1.0)])
transformed_image = transform(image=image)['image']
transformed_gt = transform(image=ground_truth)['image']
# print(f'data Typye check: \n transformed_image -> shape : {transformed_image.shape} dataytpe: {transformed_image.dtype}:: \n transformed_gt :: shape {transformed_gt.shape} dtype {transformed_gt.dtype} ')
# print()
return transformed_image, transformed_gt
def transpose(image, ground_truth):
# print('Applying Transpose')
# print(f'recive : image : {image.shape} {image.dtype} :: ground_truth { ground_truth.shape} { ground_truth.dtype} ')
transform = A.Compose([A.Transpose(p=1.0)])
transformed_image = transform(image=image)['image']
transformed_gt = transform(image=ground_truth)['image']
# print(f'data Typye check: \n transformed_image -> shape : {transformed_image.shape} dataytpe: {transformed_image.dtype}:: \n transformed_gt :: shape {transformed_gt.shape} dtype {transformed_gt.dtype} ')
# print()
return transformed_image, transformed_gt
def rotate_90(image, ground_truth):
# print('Rotating 90 degrees')
# print(f'recive : image : {image.shape} {image.dtype} :: ground_truth { ground_truth.shape} { ground_truth.dtype} ')
transform = A.Compose([A.Rotate(limit=(90, 90), p=1.0)])
transformed_image = transform(image=image)['image']
transformed_gt = transform(image=ground_truth)['image']
# print(f'data Typye check: \n transformed_image -> shape : {transformed_image.shape} dataytpe: {transformed_image.dtype}:: \n transformed_gt :: shape {transformed_gt.shape} dtype {transformed_gt.dtype} ')
# print()
return transformed_image, transformed_gt
def rotate_180(image, ground_truth):
# print('Rotating 180 degrees')
# print(f'recive : image : {image.shape} {image.dtype} :: ground_truth { ground_truth.shape} { ground_truth.dtype} ')
transform = A.Compose([A.Rotate(limit=(180, 180), p=1.0)])
transformed_image = transform(image=image)['image']
transformed_gt = transform(image=ground_truth)['image']
# print(f'data Typye check: \n transformed_image -> shape : {transformed_image.shape} dataytpe: {transformed_image.dtype}:: \n transformed_gt :: shape {transformed_gt.shape} dtype {transformed_gt.dtype} ')
# print()
return transformed_image, transformed_gt
def rotate_270(image, ground_truth):
# print('Rotating 270 degrees')
# print(f'recive : image : {image.shape} {image.dtype} :: ground_truth { ground_truth.shape} { ground_truth.dtype} ')
transform = A.Compose([A.Rotate(limit=(270, 270), p=1.0)])
transformed_image = transform(image=image)['image']
transformed_gt = transform(image=ground_truth)['image']
# print(f'data Typye check: \n transformed_image -> shape : {transformed_image.shape} dataytpe: {transformed_image.dtype}:: \n transformed_gt :: shape {transformed_gt.shape} dtype {transformed_gt.dtype} ')
# print()
return transformed_image, transformed_gt