VerificationCodeRecognition/var.py at master · MgArcher/VerificationCodeRecognition · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# !/usr/bin/env python
# -*-coding:utf-8 -*-

"""
# File       : var.py
# Time       ：2023/9/15 11:17
# Author     ：yujia
# version    ：python 3.6
# Description：
"""
import torch
from torch.utils.data import DataLoader


from utils import utils
from tool import dataloader
from tool import load, process


class Opt():
    valRoot = r"data"
    cuda = True
    pretrained = 'expr/best_expr.pth'
    alphabet_path = 'tool/charactes_keys.txt'
    batchSize = 64
    nh = 256
    nc = 3
    workers = 0
    imgH = 32
    imgW = 100
    model_name = "crnnlite"


opt = Opt()
sampler = None
if opt.cuda:
    device = torch.device('cuda')
else:
    device = torch.device('cpu')
alphabet = dataloader.get_charactes_keys(opt.alphabet_path)
test_dataset = dataloader.CaptchaDataset(opt.valRoot, [opt.imgH, opt.imgW], opt.nc)
test_loader = DataLoader(
    test_dataset, batch_size=opt.batchSize,
    shuffle=True, sampler=sampler,
    num_workers=int(opt.workers),
    )

# 加载模型
model = load.load_model(opt, alphabet, opt.model_name)
# loss
criterion = torch.nn.CTCLoss()
# 解码器
converter = utils.strLabelConverter(alphabet)
val_acc = process.val(test_loader, model, criterion, converter, device)
print("val_acc:", val_acc)