draw_keypoints¶
- torchvision.utils.draw_keypoints(image: Tensor, keypoints: Tensor, connectivity: Optional[list[tuple[int, int]]] = None, colors: Optional[Union[str, tuple[int, int, int]]] = None, radius: int = 2, width: int = 3, visibility: Optional[Tensor] = None) Tensor[原始碼]¶
在給定的 RGB 影像上繪製關鍵點。影像值應為 uint8 型別的 [0, 255] 或浮點型別 [0, 1]。可以同時繪製多個例項的關鍵點。
此方法允許根據關鍵點的可見性來繪製關鍵點及其連線。
- 引數:
image (Tensor) – 形狀為 (3, H, W) 且 dtype 為 uint8 或 float 的張量。
keypoints (Tensor) – 形狀為 (num_instances, K, 2) 的張量，表示 N 個例項中每個例項的 K 個關鍵點位置，格式為 [x, y]。
connectivity (List[Tuple[int, int]]]) – 一個元組列表，其中每個元組包含一對要連線的關鍵點。如果連線的兩個關鍵點中至少有一個的
visibility為 False，則不會繪製此特定連線。因不可見而排除的連線是按例項計算的。colors (str, Tuple) – 顏色可以表示為 PIL 字串（例如，“red”或“#FF00FF”），或作為 RGB 元組（例如，
(240, 10, 157)）。radius (int) – 表示關鍵點半徑的整數。
width (int) – 表示連線關鍵點的線的寬度的整數。
visibility (Tensor) – 形狀為 (num_instances, K) 的張量，指定 N 個例項中每個例項的 K 個關鍵點的可見性。True 表示相應的關鍵點可見並應被繪製。False 表示不可見，因此不會繪製該點或包含它的可能連線。輸入張量將被轉換為 bool 型別。預設值
None表示所有關鍵點都可見。有關更多詳細資訊，請參閱 繪製帶可見性的關鍵點。
- 返回:
繪製了關鍵點的影像張量。
- 返回型別:
img (Tensor[C, H, W])
使用
draw_keypoints的示例
