mirror of
https://github.com/we0091234/crnn_plate_recognition.git
synced 2025-10-05 10:56:50 +08:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
# -*- coding:utf-8 -*-
|
|
# Author: RubanSeven
|
|
|
|
import cv2
|
|
import imageio
|
|
import numpy as np
|
|
from augment import distort, stretch, perspective
|
|
|
|
|
|
def create_gif(image_list, gif_name, duration=0.1):
|
|
frames = []
|
|
for image in image_list:
|
|
frames.append(image)
|
|
imageio.mimsave(gif_name, frames, 'GIF', duration=duration)
|
|
return
|
|
|
|
|
|
if __name__ == '__main__':
|
|
im = cv2.imdecode(np.fromfile(r"H:\@chepai\Text-Image-Augmentation-python-master\imgs\demo.png",dtype=np.uint8),-1)
|
|
# im = cv2.resize(im, (200, 64))
|
|
cv2.imshow("im_CV", im)
|
|
distort_img_list = list()
|
|
stretch_img_list = list()
|
|
perspective_img_list = list()
|
|
for i in range(1):
|
|
distort_img = distort(im, 4)
|
|
distort_img_list.append(distort_img)
|
|
# cv2.imshow("distort_img", distort_img)
|
|
|
|
stretch_img = stretch(distort_img, 4)
|
|
# cv2.imshow("stretch_img", stretch_img)
|
|
stretch_img_list.append(stretch_img)
|
|
|
|
perspective_img = perspective(stretch_img)
|
|
cv2.imshow("perspective_img", perspective_img)
|
|
perspective_img_list.append(perspective_img)
|
|
cv2.waitKey(1)
|
|
|
|
create_gif(distort_img_list, r'imgs/distort.gif')
|
|
create_gif(stretch_img_list, r'imgs/stretch.gif')
|
|
create_gif(perspective_img_list, r'imgs/perspective.gif')
|