Update demo_RetinaFace_with_postprocess_onnx.py

PINTO0309 · Aug 5, 2023 · 83f0f34 · 83f0f34
1 parent c2778cf
commit 83f0f34
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/399_RetinaFace_MobileNetv2/demo/demo_RetinaFace_with_postprocess_onnx.py b/399_RetinaFace_MobileNetv2/demo/demo_RetinaFace_with_postprocess_onnx.py
@@ -62,7 +62,7 @@ def __init__(
     def __call__(
         self,
         image: np.ndarray,
-    ) -> Tuple[np.ndarray, np.ndarray]:
+    ) -> np.ndarray:
         """
 
         Parameters
@@ -72,11 +72,6 @@ def __call__(
 
         Returns
         -------
-        face_boxes: np.ndarray
-            Predicted face boxes: [facecount, y1, x1, y2, x2]
-
-        face_scores: np.ndarray
-            Predicted face box scores: [facecount, score]
         """
         temp_image = copy.deepcopy(image)