vincent341 / T2FGAN-pytorch

pytorch version of T2FGAN

T2FGAN-pytorch

pytorch version of T2FGAN

Using the code

1. Please download the trained model for generating underwater images by using the link.
2. Please download the pretrained model of classifier by using the link.
3. To train, run adv_train200324.py by using the command line below:

    python3 adv_train200324 --MGpath "file path in step1" --Cpath "file path in step 2"