aleju--imgaug
wehub-resource-sync
51 行
1.4 KiB
Python
51 行
1.4 KiB
Python
from __future__ import print_function, division
|
|
from itertools import cycle
|
|
|
|
import numpy as np
|
|
import cv2
|
|
|
|
import imgaug as ia
|
|
from imgaug import augmenters as iaa
|
|
|
|
POINT_SIZE = 5
|
|
DEG_PER_STEP = 2
|
|
TIME_PER_STEP = 1
|
|
|
|
|
|
def main():
|
|
image = ia.data.quokka(0.5)
|
|
height, width = image.shape[0], image.shape[1]
|
|
center_x = width // 2
|
|
center_y = height // 2
|
|
r = int(min(image.shape[0], image.shape[1]) / 3)
|
|
|
|
cv2.namedWindow("aug", cv2.WINDOW_NORMAL)
|
|
cv2.imshow("aug", image[:, :, ::-1])
|
|
cv2.waitKey(TIME_PER_STEP)
|
|
|
|
for angle in cycle(np.arange(0, 360, DEG_PER_STEP)):
|
|
rad = np.deg2rad(angle-90)
|
|
point_x = int(center_x + r * np.cos(rad))
|
|
point_y = int(center_y + r * np.sin(rad))
|
|
|
|
aug = iaa.MotionBlur(k=35, angle=angle, direction=-1.0)
|
|
img_aug = aug.augment_image(image)
|
|
img_aug[
|
|
point_y-POINT_SIZE:point_y+POINT_SIZE+1,
|
|
point_x-POINT_SIZE:point_x+POINT_SIZE+1,
|
|
:] = np.array([0, 255, 0])
|
|
|
|
aug_inv = iaa.MotionBlur(k=35, angle=angle, direction=1.0)
|
|
img_aug_inv = aug_inv.augment_image(image)
|
|
img_aug_inv[
|
|
point_y - POINT_SIZE:point_y + POINT_SIZE + 1,
|
|
point_x - POINT_SIZE:point_x + POINT_SIZE + 1,
|
|
:] = np.array([0, 255, 0])
|
|
|
|
cv2.imshow("aug", np.hstack([img_aug[:, :, ::-1], img_aug_inv[:, :, ::-1]]))
|
|
cv2.waitKey(TIME_PER_STEP)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|