zishuo/crnn_plate_recognition
1
0
Fork 0
mirror of https://github.com/we0091234/crnn_plate_recognition.git synced 2025-12-24 12:12:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
we0091234
2023-03-20 20:38:15 +08:00
parent 0d82a0a9f1
commit ffa7567057
8 changed files with 199975 additions and 62915 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -11,6 +11,7 @@ build/
result1/
output/
color_model/
pretrainde_model/
demoMy.py
get_every_provice_number.py
get_train_less_data.py

260313
datasets/train.txt
View File

File diff suppressed because it is too large Load Diff

2562
datasets/val.txt Normal file
View File

File diff suppressed because it is too large Load Diff

4
demo.py
View File

@@ -73,8 +73,8 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--model_path', type=str, default='saved_model/best.pth', help='model.pt path(s)')
# parser.add_argument('--image_path', type=str, default='images/tmp6FC6.png', help='source')
# parser.add_argument('--image_path', type=str, default='/mnt/EPan/carPlate/@realTest2_noTraining/realrealTest', help='source')
parser.add_argument('--image_path', type=str, default='/mnt/Gu/trainData/plate/new_git_train/val_verify', help='source')
parser.add_argument('--image_path', type=str, default='/mnt/EPan/carPlate/@realTest2_noTraining/realrealTest', help='source')
# parser.add_argument('--image_path', type=str, default='/mnt/Gu/trainData/plate/new_git_train/val_verify', help='source')
parser.add_argument('--img_h', type=int, default=48, help='height')
parser.add_argument('--img_w',type=int,default=168,help='width')
parser.add_argument('--LPRNet',action='store_true',help='use LPRNet') #True代表使用LPRNet ,False代表用plateNet

2
lib/config/360CC_config.yaml
View File

@@ -24,7 +24,7 @@ DATASET:
ALPHABETS: ''
TRAIN:
BATCH_SIZE_PER_GPU: 384
BATCH_SIZE_PER_GPU: 256
SHUFFLE: True
BEGIN_EPOCH: 0
END_EPOCH: 100

4
plateLabel.py
View File

@@ -20,8 +20,8 @@ def is_str_right(plate_name):
return True
if __name__=="__main__":
parser = argparse.ArgumentParser()
parser.add_argument('--image_path', type=str, default="/mnt/Gu/trainData/plate/new_git_train/CCPD_CRPD_OTHER", help='source')
parser.add_argument('--label_file', type=str, default='datasets/train.txt', help='model.pt path(s)')
parser.add_argument('--image_path', type=str, default="/mnt/EPan/carPlate/@realTest2_noTraining/realrealTest", help='source')
parser.add_argument('--label_file', type=str, default='datasets/val.txt', help='model.pt path(s)')
opt = parser.parse_args()
rootPath = opt.image_path

BIN
saved_model/best.pth
View File

Binary file not shown.

4
train.py
View File

@@ -58,8 +58,8 @@ def main():
# construct face related neural networks
#cfg =[8,8,16,16,'M',32,32,'M',48,48,'M',64,128] #small model
# cfg =[16,16,32,32,'M',64,64,'M',96,96,'M',128,256]#medium model
cfg =[32,32,64,64,'M',128,128,'M',196,196,'M',256,256] #big model
cfg =[16,16,32,32,'M',64,64,'M',96,96,'M',128,256]#medium model
# cfg =[32,32,64,64,'M',128,128,'M',196,196,'M',256,256] #big model
# model = crnn.get_crnn(config,cfg=cfg)
model = myNet_ocr(num_classes=len(plate_chr),cfg=cfg)
# model = build_lprnet(num_classes=len(plate_chr))