Блог

Я запускаю CycleGAN с разными типами tiff в trainA и trainB. Размер tiff составляет 256×256 пикселей и имеет 1 канал на пиксель. Я использую tiff для получения широкого диапазона значений.

Я изменил код, как предложено в репозитории pytorch-CycleGAN-and-pix2pix (https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/issues/320 и подобное), но то, что я получил во время обучения ./checkpoints , — это трехканальные PNG. Как вы думаете, можно ли изменить код так, чтобы он переходил с 1-канального tiff на 1-канальный tiff без потери информации? Насколько я понимаю, в настоящее время код преобразует импортированные файлы в PNG по пути. Другими словами: я хотел бы, чтобы мои тензоры были [256*256*int_range,1] . Спасибо за помощь!

Вы пробовали добавлять параметры —input_nc 1 —output_nc 1 во время обучения? Это преобразует количество каналов из 3 в 1.