draw_box

draw_box(img: np.ndarray, box: _Box, color: _Color = (0, 255, 0), thickness: _Thickness = 2) -> np.ndarray

• 依賴

  • 繪圖功能為可選依賴；請先安裝：

    pip install "capybara-docsaid[visualization]"

• 說明：在影像上繪製 Bounding Box。

• 參數

  • img (np.ndarray)：要繪製的影像，為 NumPy 陣列。
  • box (Union[Box, np.ndarray])：要繪製的 Bounding Box，可以是 Box 物件或 NumPy 陣列形式的 [x1, y1, x2, y2]。
  • color (_Color)：要繪製的框的顏色（OpenCV 慣例：BGR）。預設為 (0, 255, 0)。
  • thickness (_Thickness)：要繪製的框線的粗細。預設為 2。

• 傳回值

  • np.ndarray：繪製了框的影像。

• 範例

  from capybara import Box, imread
  from capybara.vision.visualization.draw import draw_box
  
  img = imread('lena.png')
  box = Box([20, 20, 100, 100])
  box_img = draw_box(img, box, color=(0, 255, 0), thickness=2)