【文件属性】：
文件名称：OT_CycleGAN:Optimal Transport-driven CycleGAN（OT-CycleGAN）的实现
文件大小：6KB
文件格式：ZIP
更新时间：2021-05-29 10:42:52
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)