거리 기반 가중치를 적용한 손실함수를 활용한 영상 분할 알고리즘의 경계 예측 개선

초록

본 논문에서는 실시간 영상 분할 모델에서 인접한 객체 간 경계를 명확히 구분하지 못하는 문제를 해결하기 위해, 거리 기반의 형태 손실함수를 제안하였다. 제안한 손실함수를 사용하여 모델을 학습시킨 결과, 기존 교차 엔트로피 손실함수를 적용했을 때보다 정답에 더 근접한 형태로 객체를 분할할 수 있음을 실험적으로 확인하였다.

키워드