Skip to content

Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation

License

Notifications You must be signed in to change notification settings

chenxu31/UGATIT-pytorch

 
 

Repository files navigation

code

preprocessing

download code to Lian_project_code/

train U-GAT-IT

python main_pelvic.py --gpu {GPU_ID} --data_dir {DATA_DIR} --checkpoint_dir {CHECKPOINT_DIR} --light True

test U-GAT-IT

python main_pelvic.py --phase test --gpu {GPU_ID} --data_dir {DATA_DIR} --checkpoint_dir {CHECKPOINT_DIR} --result_dir {OUTPUT_DIR} --light True

About

Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%