zishuo/crnn_plate_recognition
1
0
Fork 0
mirror of https://github.com/we0091234/crnn_plate_recognition.git synced 2025-10-06 03:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

demo

Browse Source
This commit is contained in:
we0091234
2022-11-13 21:07:55 +08:00
parent e57ccc4cc8
commit cce10c303b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
demo.py
View File

@@ -61,7 +61,7 @@ def init_model(device,model_path):
check_point = torch.load(model_path,map_location=device) check_point = torch.load(model_path,map_location=device)
model_state=check_point['state_dict'] model_state=check_point['state_dict']
cfg = check_point['cfg'] cfg = check_point['cfg']
model = myNet_ocr(num_classes=78,export=True,cfg=cfg) #export True 用来推理 model = myNet_ocr(num_classes=len(plate_chr),export=True,cfg=cfg) #export True 用来推理
# model =build_lprnet(num_classes=len(plate_chr),export=True) # model =build_lprnet(num_classes=len(plate_chr),export=True)
model.load_state_dict(model_state) model.load_state_dict(model_state)
model.to(device) model.to(device)