wadawdawdawddwa
Browse files
main.py
CHANGED
@@ -65,7 +65,7 @@ async def predict_image(file: UploadFile = File(...)):
|
|
65 |
if results[i]["label"] == "background":
|
66 |
continue
|
67 |
mask_area = calculate_mask_area(results[i]["mask"])
|
68 |
-
print(f"{results[i][
|
69 |
score = mask_area / (500 * 500 - background_area)
|
70 |
results[i]["score"] = score
|
71 |
|
|
|
65 |
if results[i]["label"] == "background":
|
66 |
continue
|
67 |
mask_area = calculate_mask_area(results[i]["mask"])
|
68 |
+
print(f"{results[i]['label']} area: {mask_area}")
|
69 |
score = mask_area / (500 * 500 - background_area)
|
70 |
results[i]["score"] = score
|
71 |
|