taesungp / contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)

Home Page:https://taesung.me/ContrastiveUnpairedTranslation/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multi-channel images as inputs

LiangJunhao-THU opened this issue · comments

Hello,
Great work!

I would like to ask if CUT can handle multi-channel images?
The RGB image can be regarded as a 3-channel image, but now I need to use a 100-channel image as the input of CUT.
Can the current model directly accomplish this task? Or what code do I need to modify?

Thanks