Pytorch 具有单通道TIFF作为输入和输出的CycleGAN

Pytorch 具有单通道TIFF作为输入和输出的CycleGAN,pytorch,generative-adversarial-network,Pytorch,Generative Adversarial Network,我在trainA和trainB中与不同类型的TIFF一起运行CycleGAN。TIFF大小为256x256像素,每个像素有一个通道。我使用TIFF具有广泛的价值观 我按照中的建议(以及类似建议)更改了代码,但在/checkpoints中的培训中,我得到的是三个通道PNG。您认为有可能更改代码,使其从1通道tiff变为1通道tiff而不丢失信息吗?据我所知,目前代码正在将导入的文件转换为PNG。换句话说:我希望我的张量是[256*256*int\u范围,1]。谢谢你的帮助 您是否尝试过在培训期间添

我在trainA和trainB中与不同类型的TIFF一起运行CycleGAN。TIFF大小为256x256像素,每个像素有一个通道。我使用TIFF具有广泛的价值观


我按照中的建议(以及类似建议)更改了代码,但在
/checkpoints
中的培训中,我得到的是三个通道PNG。您认为有可能更改代码,使其从1通道tiff变为1通道tiff而不丢失信息吗?据我所知,目前代码正在将导入的文件转换为PNG。换句话说:我希望我的张量是
[256*256*int\u范围,1]
。谢谢你的帮助

您是否尝试过在培训期间添加参数--input_nc 1--output_nc 1?它会将通道数从3转换为1