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 影像上繪製關鍵點。影像值應為 [0, 255] 範圍內的 uint8 型別或 [0, 1] 範圍內的 float 型別。可以一次為多個例項繪製關鍵點。
此方法允許根據關鍵點的可見性繪製關鍵點及其連線線。
- 引數：
image (Tensor) – 形狀為 (3, H, W)，資料型別為 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的示例
