This is the official Pytorch implementation of ColorMamba

About assets/autoencoder.pth：https://github.com/theonegis/ganstfm/tree/main/assets

Pre-trained parameters, please import them into "models/.G_pre.pt" for training：https://drive.google.com/file/d/1TBkao5mZy8OqXnjoD98H-vmITfiathZA/view?usp=drive_link

2020VCIP Datasets：https://drive.google.com/file/d/1_iI8kaTbMuYnWGll5H294YTt78qcX5Lg/view