Segmentation mask의 center point 계산하기

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를 계산할지 고민하다가 위 처럼 짜보았는데, 혹시 도움이 필요하신 분이 계실까봐 공유합니다