LabelOneHot #Label #One-Hot #Segmentation #FaceParsing #FocalLoss #DRO #loss1 [Python] Segmentation 분야에서 클래스 별 레이블이 필요할 때 - Label One-Hot Encoding / python 코드 Segmentation 분야에서 딥러닝 모델을 학습시킬 때, 레이블에는 주로 아래 사진과 같이 한 장의 이미지에 픽셀 별로 Segmentation된 이미지가 들어간다. 하지만 모델을 고치거나 Loss Function을 변형하려 하면 각각의 클래스들로만 구성된 레이어들이 필요할 때가 있다. 나는 Fairness Learning을 위해 DRO를 사용하거나, Focal Loss Function을 사용하기 위해 필요했다. 예를 들어 클래스가 왼쪽 눈, 오른쪽 눈, ... , 목 으로 총 19개 있다면 왼쪽 눈만 있는 레이어, 오른쪽 눈만 있는 레이어, ... , 목만 있는 레이어 해서 총 19개의 레이어가 필요한 것이다. 사진을 보고 예시를 들자면 아래와 같은 레이어들이 필요한 것이다. Label One-Hot.. 2023. 2. 2. 이전 1 다음