mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-10-06 22:50:14 +02:00
updating:rotation augmentation is provided for machine based reading order
This commit is contained in:
parent
825b2634f9
commit
dd21a3b33a
1 changed files with 13 additions and 1 deletions
|
@ -356,6 +356,18 @@ def rotate_max_area(image, rotated, rotated_label, angle):
|
|||
x2 = x1 + int(wr)
|
||||
return rotated[y1:y2, x1:x2], rotated_label[y1:y2, x1:x2]
|
||||
|
||||
def rotate_max_area_single_image(image, rotated, angle):
|
||||
""" image: cv2 image matrix object
|
||||
angle: in degree
|
||||
"""
|
||||
wr, hr = rotatedRectWithMaxArea(image.shape[1], image.shape[0],
|
||||
math.radians(angle))
|
||||
h, w, _ = rotated.shape
|
||||
y1 = h // 2 - int(hr / 2)
|
||||
y2 = y1 + int(hr)
|
||||
x1 = w // 2 - int(wr / 2)
|
||||
x2 = x1 + int(wr)
|
||||
return rotated[y1:y2, x1:x2]
|
||||
|
||||
def rotation_not_90_func(img, label, thetha):
|
||||
rotated = imutils.rotate(img, thetha)
|
||||
|
@ -365,7 +377,7 @@ def rotation_not_90_func(img, label, thetha):
|
|||
|
||||
def rotation_not_90_func_single_image(img, thetha):
|
||||
rotated = imutils.rotate(img, thetha)
|
||||
return rotate_max_area(img, rotated, thetha)
|
||||
return rotate_max_area_single_image(img, rotated, thetha)
|
||||
|
||||
|
||||
def color_images(seg, n_classes):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue