loss를 OCRnet에서 어떤 방식으로 계산하는 걸까요?

[kmouleejunhyuk]

shape가 아예 다른데, 픽셀 별로 onehot하는 걸까요?
만약 그렇다면 mask에 "가려지는"(경재님이 말씀하신 mask label 순서 문제) 부분은, 겹쳐 있으면 어떻게 되는 걸까요?
ex: 한 픽셀의 seg 결과
pred: 00000000001
mask(gt): 5, 11

[jiwoo0212]

위 코드는 pytorch dataset에서 mask를 만드는 부분인데
겹쳐지더라도 결국엔 한 픽셀당 하나의 값으로 결정되지 않나요? 🤔

[kmouleejunhyuk]

맞는 것 같습니다

[jaehwan-AI]

1. 어느 부분의 shape이 의문이신가요?
2. OCRNet의 loss는 RMI Loss를 이용해서 계산합니다.
   

[kmouleejunhyuk]

mask와 clsout의 차원이 1, 11로 서로 상이해서 의문이 들었습니다.

[jaehwan-AI]

아 output은 argmax로 mask 차원과 동일하게 될거에요