OT_CycleGAN:Optimal Transport-driven CycleGAN(OT-CycleGAN)的实现

时间:2024-06-18 04:29:32
【文件属性】:

文件名称:OT_CycleGAN:Optimal Transport-driven CycleGAN(OT-CycleGAN)的实现

文件大小:6KB

文件格式:ZIP

更新时间:2024-06-18 04:29:32

Python

OT-CycleGAN(最优传输驱动的 CycleGAN) 用于逆问题无监督学习的最佳传输驱动 CycleGAN 的Pytorch 实现 [] 在自定义数据集上训练 python train.py --data_path directory_for_custom_dataset --domain_A subdirctory_for_domain_A \\ --domain_B subdirectory_for_domain_B --extension png 实施细则 data_path应该包含domain_A和domain_B作为其子目录。 扩展应该能够用numpy.load或SimpleITK.ReadImage打开。 如果这些函数不支持您的文件扩展名,请修改 [ ]。 noramlize仅支持minmax、tanh、CT、None 之一。 如果


【文件预览】:
OT_CycleGAN-master
----train.py(9KB)
----config.py(3KB)
----utils.py(2KB)
----models.py(6KB)
----README.md(2KB)

网友评论