fix (#22)

ternaus · Aug 31, 2020 · a4e3433 · a4e3433
1 parent b731f4b
commit a4e3433
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/retinaface/__init__.py b/retinaface/__init__.py
@@ -1 +1 @@
-__version__ = "0.0.6"
+__version__ = "0.0.7"
diff --git a/retinaface/utils.py b/retinaface/utils.py
@@ -54,5 +54,5 @@ def vis_annotations(image: np.ndarray, annotations: List[Dict[str, Any]]) -> np.
         x_min = np.clip(x_min, 0, x_max - 1)
         y_min = np.clip(y_min, 0, y_max - 1)
 
-        vis_image = cv2.rectangle(image, (x_min, y_min), (x_max, y_max), color=(0, 255, 0), thickness=2)
+        vis_image = cv2.rectangle(vis_image, (x_min, y_min), (x_max, y_max), color=(0, 255, 0), thickness=2)
     return vis_image