fix JSON serialization error caused by conf=nan

master
huangsx 12 months ago
parent 6c902c9af9
commit 2b7f3c3548

@ -108,6 +108,7 @@ async def vehicle_license_plate_recognition(file: List[UploadFile] = File(...)):
plates = catcher(img) plates = catcher(img)
results = list() results = list()
for code, conf, plate_type, box in plates: for code, conf, plate_type, box in plates:
if "nan" != f"{conf}": # conf=nan会导致Json序列化错误
plate = dict(code=code, conf=float(conf), plate_type=type_list[plate_type], box=box) plate = dict(code=code, conf=float(conf), plate_type=type_list[plate_type], box=box)
results.append(plate) results.append(plate)
return BaseResponse().http_ok_response({'plate_list': results}) return BaseResponse().http_ok_response({'plate_list': results})

Loading…
Cancel
Save