search

daintlab / CRAFT-Reimplementation

CRAFT-Pytorch：Character Region Awareness for Text Detection Reimplementation for Pytorch
0 stars 0 forks source link

hard negative mining 부분 #11

Closed beopst closed 2 years ago

beopst commented 2 years ago

https://github.com/daintlab/CRAFT-Reimplementation/blob/686863f80d973cd5ecfdbae97c250514e22fa6a4/loss/mseloss.py#L52-L56

이 부분을 보면 confidence mask를 먼저 곱하는 것 같은데 그러면 hard negative 뽑는 부분에서 문제가 되지 않나요?

beopst commented 2 years ago

https://github.com/daintlab/CRAFT-Reimplementation/blob/686863f80d973cd5ecfdbae97c250514e22fa6a4/loss/mseloss.py#L32-L34

이 부분에서 confidence mask가 곱해진 걸 가지고 topk를 뽑는데 그럼 negative 부분은 그냥 loss가 다 0으로 박혀 있는게 아닌가 싶습니다.

beopst commented 2 years ago

아 negative 부분의 default confidence 값이 1이군요. 이슈 클로징할께요 ㅋㅋ