mirror of
https://github.com/we0091234/crnn_plate_recognition.git
synced 2025-09-26 23:45:51 +08:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import random
|
|
# https://albumentations.ai/docs/examples/example/
|
|
# https://github.com/albumentations-team/albumentations
|
|
import cv2
|
|
# from matplotlib import pyplot as plt
|
|
import numpy as np
|
|
import os
|
|
|
|
import albumentations as A
|
|
|
|
def allFileList(rootfile,allFile):
|
|
folder =os.listdir(rootfile)
|
|
for temp in folder:
|
|
fileName = os.path.join(rootfile,temp)
|
|
if os.path.isfile(fileName):
|
|
allFile.append(fileName)
|
|
else:
|
|
allFileList(fileName,allFile)
|
|
|
|
|
|
fileList = []
|
|
rootPath = r"E:\carPlate\trainAug\1"
|
|
savePath =r"E:\carPlate\trainAug\2"
|
|
allFileList(rootPath,fileList)
|
|
i = 0
|
|
for temp in fileList:
|
|
i = i+1
|
|
print(i,temp)
|
|
image = cv2.imdecode(np.fromfile(temp,dtype=np.uint8),-1)
|
|
# image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
|
transform = A.Compose([
|
|
|
|
A.Blur(blur_limit=20,p=1),
|
|
# A.MedianBlur(),
|
|
# A.ElasticTransform(p=1),
|
|
# # # A.RandomBrightnessContrast(),
|
|
# A.OpticalDistortion(),
|
|
# A.MedianBlur(blur_limit=8,p=1),
|
|
A.RandomBrightnessContrast(p=1),
|
|
# A.ImageCompression(),
|
|
# A.RGBShift(),
|
|
# A.RandomGamma(),
|
|
# A.VerticalFlip(),
|
|
# A.Rotate()
|
|
# A.GridDistortion(),
|
|
# A.HueSaturationValue(),
|
|
])
|
|
random.seed(79)
|
|
augmented_image = transform(image=image)['image']
|
|
imageName = temp.split("\\")[-1]
|
|
# cv2.imshow("haha1",image)
|
|
# cv2.imshow("haha2", augmented_image)
|
|
# cv2.waitKey(0)
|
|
savePicPath = os.path.join(savePath,imageName)
|
|
cv2.imencode('.jpg', augmented_image)[1].tofile(savePicPath)
|
|
|