mirror of
https://github.com/we0091234/crnn_plate_recognition.git
synced 2025-09-26 23:45:51 +08:00
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import cv2
|
|
import imageio
|
|
import numpy as np
|
|
import os
|
|
import shutil
|
|
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)
|
|
|
|
|
|
rootPath = r"D:\trainTemp\carPlate\train"
|
|
palteStr=r"#京沪津渝冀晋蒙辽吉黑苏浙皖闽赣鲁豫鄂湘粤桂琼川贵云藏陕甘青宁新学警港澳挂使领民深危险品0123456789ABCDEFGHJKLMNPQRSTUVWXYZ"
|
|
labelFile =r"D:\trainTemp\carPlate\0914_train.txt"
|
|
plateDict ={}
|
|
for i in range(len(list(palteStr))):
|
|
plateDict[palteStr[i]]=i
|
|
fp = open(labelFile,"w")
|
|
file =[]
|
|
allFileList(rootPath,file)
|
|
picNum = 0
|
|
for jpgFile in file:
|
|
jpgName = jpgFile.split("\\")[-1]
|
|
name =jpgName.split("_")[0]
|
|
labelStr=" "
|
|
strList = list(name)
|
|
# print(jpgFile)
|
|
for i in range(len(strList)):
|
|
labelStr+=str(plateDict[strList[i]])+" "
|
|
while i<7:
|
|
labelStr+=str(0)+" "
|
|
i+=1
|
|
picNum+=1
|
|
print(jpgFile+labelStr)
|
|
fp.write(jpgFile+labelStr+"\n")
|