121 lines
3.3 KiB
Python
121 lines
3.3 KiB
Python
import json
|
|
|
|
|
|
class Result(object):
|
|
def __init__(self) -> None:
|
|
# self.res_dict = {
|
|
# "合格证编号": "",
|
|
# "发证日期":"",
|
|
# "车辆制造企业名称":"",
|
|
# "车辆品牌":"",
|
|
# "车辆名称":"",
|
|
# "车辆型号":"",
|
|
# "车架号":"",
|
|
# "车身颜色":"",
|
|
# "底盘型号":"",
|
|
# "底盘ID":"",
|
|
# "底盘合格证编号":"",
|
|
# "发动机型号":"",
|
|
# "发动机号":"",
|
|
# "燃料种类":"",
|
|
# "排量":"",
|
|
# "功率":"",
|
|
# "排放标准":"",
|
|
# "油耗":"",
|
|
# "外廓尺寸1":"",
|
|
# "外廓尺寸2":"",
|
|
# "外廓尺寸3":"",
|
|
# "货箱内部尺寸1":"",
|
|
# "货箱内部尺寸2":"",
|
|
# "货箱内部尺寸3":"",
|
|
# "钢板弹簧片数":"",
|
|
# "轮胎数":"",
|
|
# "轮胎规格":"",
|
|
# "前轮距":"",
|
|
# "后轮距":"",
|
|
# "轴距":"",
|
|
# "轴荷":"",
|
|
# "轴数":"",
|
|
# "转向形式":"",
|
|
# "总质量":"",
|
|
# "整备质量":"",
|
|
# "额定载质量":"",
|
|
# "载质量利用系数":"",
|
|
# "准牵引总质量":"",
|
|
# "半挂车鞍座最大允许总质量":"",
|
|
# "驾驶室准乘人数":"",
|
|
# "额定载客":"",
|
|
# "最高设计车速":"",
|
|
# "车辆制造日期":"",
|
|
# }
|
|
|
|
self.res_dict = {
|
|
"HGZBH": "",
|
|
"FZRQ":"",
|
|
"CLZZQYMC":"",
|
|
"CLPP":"",
|
|
"CLMC":"",
|
|
"CLXH":"",
|
|
"CJH":"",
|
|
"CSYS":"",
|
|
"DPXH":"",
|
|
"DPID":"",
|
|
"DPHGZBH":"",
|
|
"FDJXH":"",
|
|
"FDJH":"",
|
|
"RLZL":"",
|
|
"PL":"",
|
|
"GL":"",
|
|
"PFBZ":"",
|
|
"YH":"",
|
|
"WKCC1":"",
|
|
"WKCC2":"",
|
|
"WKCC3":"",
|
|
"HXNBCC1":"",
|
|
"HXNBCC2":"",
|
|
"HXNBCC3":"",
|
|
"GBTHPS":"",
|
|
"LTS":"",
|
|
"LTGG":"",
|
|
"QLJ":"",
|
|
"HLJ":"",
|
|
"ZJ":"",
|
|
"ZH":"",
|
|
"ZS":"",
|
|
"ZXXS":"",
|
|
"ZZL":"",
|
|
"ZBZL":"",
|
|
"EDZZL":"",
|
|
"ZZLLYXS":"",
|
|
"ZQYZZL":"",
|
|
"BGCAZZDYXZZL":"",
|
|
"JSSZCRS":"",
|
|
"EDZK":"",
|
|
"ZGSJCS":"",
|
|
"ZLZZRQ":"",
|
|
}
|
|
|
|
def update(self, ocr_res):
|
|
if ocr_res is not None:
|
|
# boxes, keys, txts, scores = list(zip(*ocr_res))
|
|
for box, key, text, score in ocr_res:
|
|
self.res_dict[key] = text
|
|
|
|
# if len(ocr_res)==len(self.res_dict):
|
|
# for key, (text, score) in zip(self.res_dict.keys(), ocr_res):
|
|
# if score < 0.5:
|
|
# continue
|
|
# self.res_dict[key] = text
|
|
|
|
def clear(self):
|
|
for key in self.res_dict.keys():
|
|
self.res_dict[key] = ""
|
|
|
|
|
|
def get(self):
|
|
|
|
final_res = {"code":"1",
|
|
"msg":"识别成功",
|
|
"data":self.res_dict}
|
|
|
|
return final_res |