mirror of
https://github.com/we0091234/crnn_plate_recognition.git
synced 2025-09-26 23:45:51 +08:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import os
|
|
import shutil
|
|
import numpy as np
|
|
|
|
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)
|
|
|
|
def getPlateName(plate):
|
|
jpgName = plate.split("\\")[-1]
|
|
plateName=jpgName.split("_")[0]
|
|
return plateName,plate
|
|
filePath1 = r"E:\carPlate\realrealTest"
|
|
filePath2=r"D:\trainTemp\carPlate\train\train_ori"
|
|
savePath = r"E:\carPlate\realrealTest\val"
|
|
|
|
fileList1=[]
|
|
filelist2=[]
|
|
|
|
allFileList(filePath1,fileList1)
|
|
allFileList(filePath2,filelist2)
|
|
|
|
platelist1=[]
|
|
platelist2=[]
|
|
|
|
for temp in filelist2:
|
|
platelist2.append(getPlateName(temp)[0])
|
|
for temp in fileList1:
|
|
platelist1.append(getPlateName(temp))
|
|
platelist2=list(set(platelist2))
|
|
i = 0
|
|
print(platelist2)
|
|
for temp1 in platelist1:
|
|
if temp1[0] in platelist2:
|
|
print(temp1[0],temp1[1])
|
|
folder = temp1[1].split("\\")[-2]
|
|
folderPath =os.path.join(savePath,folder)
|
|
if not os.path.exists(folderPath):
|
|
os.mkdir(folderPath)
|
|
picName=temp1[1].split("\\")[-1]
|
|
picPath =os.path.join(folderPath,picName)
|
|
shutil.move(temp1[1],picPath)
|
|
i+=1
|
|
|
|
print(i) |