Merge pull request #400 from hohaiuhsx/master
fix JSON serialization error caused by conf=nan
This commit is contained in:
@@ -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})
|
||||||
|
|||||||
Reference in New Issue
Block a user