반응형
Segmentation mask의 center point를 얻어오는게 필요해서 코드를 짜 보았습니다.
mask는 0 또는 1의 값을 갖고 있으므로 numpy 혹은 pytorch의 nonzero 함수를 사용사용하면 됩니다.
coordinates = np.nonzero(mask)
x_coordinates = coordinates[0]
y_coordinates = coordinates[1]
x_min, x_max, y_min, y_max = np.min(x_coordinates), np.max(x_coordinates), np.min(y_coordinates), np.max(y_coordinates)
x_center, y_center = (x_max + x_min) / 2, (y_max + y_min) / 2
어떻게 center point를 계산할지 고민하다가 위 처럼 짜보았는데, 혹시 도움이 필요하신 분이 계실까봐 공유합니다
반응형
'Python > PyTorch 공부' 카테고리의 다른 글
| 폴리곤을 segmentation mask로 변환하기(Polygon to mask) (0) | 2022.12.23 |
|---|---|
| segmentation mask 덩어리 갯수 확인하기 (0) | 2022.08.21 |
| [PyTorch] CLIP의 text encoder에는 attention mask가 존재합니다. (0) | 2022.08.01 |
| [PyTorch] Multi_head_attention에서 target sequence length와 source sequence length 의미 (0) | 2022.07.27 |
| [PyTorch] Tensor.retain_grad() (0) | 2022.07.19 |