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[source]¶
在給定的 RGB 圖像上繪製關鍵點。圖像的值應該是 [0, 255] 範圍內的 uint8 或 [0, 1] 範圍內的 float。可以一次繪製多個實例的關鍵點。
此方法允許根據關鍵點的可見性來繪製關鍵點及其連接。
- 參數:
image (Tensor) – 形狀為 (3, H, W) 且 dtype 為 uint8 或 float 的 Tensor。
keypoints (Tensor) – 形狀為 (num_instances, K, 2) 的 Tensor，代表每個實例的 K 個關鍵點位置，格式為 [x, y]。
connectivity (List[Tuple[int, int]]]) – 一個 List of tuple，其中每個 tuple 包含一對要連接的關鍵點。如果兩個相連的關鍵點中至少有一個的
visibility為 False，則不繪製此特定連接。因不可見性導致的排除是針對每個實例計算的。colors (str, Tuple) – 顏色可以用 PIL 字串表示，例如 “red” 或 “#FF00FF”，或者用 RGB tuple 表示，例如
(240, 10, 157)。radius (int) – 表示關鍵點半徑的整數。
width (int) – 表示連接關鍵點的線條寬度的整數。
visibility (Tensor) – 形狀為 (num_instances, K) 的 Tensor，指定每個實例的 K 個關鍵點的可見性。True 表示相應的關鍵點是可見的，應該被繪製。False 表示不可見，因此不會繪製該點或包含該點的可能連接。輸入的 tensor 將被轉換為 bool。預設值
None表示所有關鍵點都是可見的。更多詳細資訊，請參閱 使用可見性繪製關鍵點。
- 返回:
繪製了關鍵點的圖像 Tensor。
- 返回類型:
img (Tensor[C, H, W])
使用
draw_keypoints的範例
