xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/feature/add_draw_image_docs' into feature/add_draw_image_docs

This commit is contained in:
SkalskiP 2023-10-18 22:49:44 +02:00
parent 893a19a991 b35a6161dd
commit caf504c0c9
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
supervision/draw/utils.py
View File

@ -172,17 +172,20 @@ def draw_text(
def draw_image(
scene: np.ndarray, image: Union[str, np.ndarray], opacity: float, rect: Rect
scene: np.ndarray, image: Union[str, np.ndarray], opacity: float, rect: Rect
) -> np.ndarray:
"""
Draws an image onto a given scene with specified opacity and dimensions.
"""
# Input checks
assert 0.0 <= opacity <= 1.0, "Opacity has to be between 0.0 and 1.0."
assert rect.x >= 0 and rect.y >= 0, "Top left coordinates of the rect must be positive."
assert (
rect.x >= 0 and rect.y >= 0
), "Top left coordinates of the rect must be positive."
assert rect.width > 0 and rect.height > 0, "Rect dimensions should be positive."
assert rect.x + rect.width <= scene.shape[1] and rect.y + rect.height <= \
scene.shape[0], "Image exceeds scene bounds."
assert (
rect.x + rect.width <= scene.shape[1] and rect.y + rect.height <= scene.shape[0]
), "Image exceeds scene bounds."
# Read image if a path is provided
if isinstance(image, str):
@ -203,13 +206,13 @@ def draw_image(
alpha_channel = cv2.convertScaleAbs(alpha_channel * opacity)
# Get ROI from the scene
scene_roi = scene[rect.y:rect.y + rect.height, rect.x:rect.x + rect.width]
scene_roi = scene[rect.y : rect.y + rect.height, rect.x : rect.x + rect.width]
# Alpha blending
blended_roi = cv2.addWeighted(scene_roi, 1 - (alpha_channel / 255.0), image,
(alpha_channel / 255.0), 0)
# Apply the blended ROI to the scene
scene[rect.y:rect.y + rect.height, rect.x:rect.x + rect.width] = blended_roi
scene[rect.y : rect.y + rect.height, rect.x : rect.x + rect.width] = blended_roi
return scene