Discriminator 고도화 about spade_colorization HOT 7 OPEN

discriminator 의 아웃풋 로깅

from spade_colorization.

Discriminator의 input으로 LAB->RGB를 넣어보는게 좋을 듯

from spade_colorization.

아니면 discriminator 의 구조가 너무 약할 수 도 있음. generation 같은 경우엔 훨씬 못한걸 구분하기 쉽지만 colorization 은 L 이 주어지기 때문에. 실제로 video colorization 은 Self attention layer 를 사용했음.

from spade_colorization.

Discriminator의 input으로 LAB->RGB를 넣어보는게 좋을 듯

이는 실험했는데 잘 안돼서 폐기.
D의 구조를 Self-attention 쓰는 것으로 바꾸자

from spade_colorization.

rgb vs lab discriminator

target only 와 같게 spade + decoder + reconstruction only 로만 학습했는데 기존 LAB discriminator (동일한 셋팅) 보다 안나옴

wandb 결과

from spade_colorization.

Video colorization 처럼 SA layer 를 넣어서 discriminator 를 돌려주면 discriminator 가 더 잘할 수도 있음. 지금은 결과가 망해도 discriminator 가 구별을 못하니까 L GT 가 주어져서 기존 image synthesis 보다 discriminator 의 task 가 더 어려울 것 같다.

from spade_colorization.

@DongHwanJang 지금 vid colorization 의 SA discriminator 구현 된게 있는데 그건 아웃풋타입이 multiscalediscriminator 랑 달라. 이걸 multiscale discriminator 처럼 맞춰줘야 할거야. 그리고 중간 피쳐 뽑는 기능도 추가하고.

from spade_colorization.