Fix Prediction.confidence should be float but not numpy.ndarray #7

Pandede · web-flow · commit 054edbfd8164 · 2023-09-18T14:15:24.000+08:00
diff --git a/wpodnet/backend.py b/wpodnet/backend.py
@@ -120,5 +120,5 @@ def predict(self, image: Image.Image) -> Prediction:
         return Prediction(
             image=image,
             bounds=bounds.astype(np.int32),
-            confidence=max_prob
+            confidence=max_prob.item()
         )

Original file line number	Diff line number	Diff line change
`@@ -120,5 +120,5 @@ def predict(self, image: Image.Image) -> Prediction:`
`120`	`120`	`return Prediction(`
`121`	`121`	`image=image,`
`122`	`122`	`bounds=bounds.astype(np.int32),`
`123`		`- confidence=max_prob`
	`123`	`+ confidence=max_prob.item()`
`124`	`124`	`)`