search

DongHwanJang / SPADE_Colorization

Other
2 stars 0 forks source link

LAB tensor range problem #15

Closed DongHwanJang closed 4 years ago

DongHwanJang commented 4 years ago

LAB tensor가 0~255 -> 0~1 로 매핑되는데, 이 때 AB를 0~1로 scale한 값을 실제로 찍어주면 아래 그림에서 새그림 부분처럼 된다. image

즉 0.5 지점에서 discontinuity가 발생한다. 이러한 점은 loss 계산시에 큰 문제가 될 수 있다.

  1. 0과 1의 색 값은 실제로는 거의 동등하나, 수치적으로는 큰 차이가 나기 때문에, gradient 업데이트시 불안정 할 수 있다.
  2. 마찬가지로 0.5 경계에서의 값은 거의 동일하지만 인간의 눈으로 봤을 때는 정말 큰 차이가 나기 때문에, 네트워크는 로스가 충분히 작기 때문에 괜찮다고 내버려 둬도, 실제 사람이 봤을 때는 굉장히 이상한 결과물이 나올 수 있다.

이에 따라 새로운 mapping function이 필요할 것으로 보인다.

DongHwanJang commented 4 years ago

image 위와 같이 나오게 LAB 수정 (위 그림은 x, y축으로 각각 값을 -1~1로 변화시킨 것)

ThisIsIsaac commented 4 years ago

까치가 너무 아름답습니다... :heart:

@DongHwanJang 왼쪽 위가 -1,-1 이고 오른쪽 아래가 1,1 인거죠? A 가 X 축이고 B 가 Y 축으로

DongHwanJang commented 4 years ago

@ThisIsIsaac ㅋㅋ 맞아요 값은 -1에서 1로 단 Y축이 A, X축이 B