diff --git a/src/main.py b/src/main.py index fc4dcbc..35003a6 100644 --- a/src/main.py +++ b/src/main.py @@ -47,9 +47,9 @@ async def infer(image: UploadFile = File(...)) -> dict: } } """ - # 读取图片内容(转灰度,与 diagnose.py 对文件路径的处理逻辑一致) + # 读取图片内容 contents = await image.read() - img = Image.open(io.BytesIO(contents)).convert("L") + img = Image.open(io.BytesIO(contents)).convert("RGB") img_array = np.array(img) diagnoser = get_diagnoser()