T2FGAN-pytorch

pytorch version of T2FGAN

Using the code

Please download the trained model for generating underwater images by using the link.
Please download the pretrained model of classifier by using the link.
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"

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
.gitattributes		.gitattributes
Fusion_model.py		Fusion_model.py
Lf_Generator.py		Lf_Generator.py
LightmarkGenerator.py		LightmarkGenerator.py
Myclassifier.py		Myclassifier.py
README.md		README.md
adv_train200324.py		adv_train200324.py
model.py		model.py
test.png		test.png
utils.py		utils.py