具有1通道tiff作为输入和输出的CycleGAN

Question

我正在使用trainA和trainB中不同类型的tiff来运行CycleGAN。 tiff的大小为256x256像素，每个像素有1个通道。我正在使用tiff来具有多种值。

我按照pytorch-CycleGAN-and-pix2pix repo（https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/issues/320和类似的内容）中的建议更改了代码，但是在./checkpoints的培训过程中发现的是三通道PNG。您是否认为可以更改代码，使其从1通道tiff变为1通道tiff，而不会丢失信息？据我了解，目前代码正在将导入的文件一路转换为PNG。换句话说：我希望我的张量为[256*256*int_range,1]。感谢您的帮助！

Answer 1

您是否在训练期间尝试添加参数--input_nc 1 --output_nc 1？它将通道数从3转换为1。