the type dataset of gray

[lzzlxxlsz]

my dataset is gray picture.and I only set :
self.parser.add_argument('--nc', type=int, default=1, help='input image channels')

but I run the train.py ,in the model.py:
def train_one_epoch(self):
""" Train the model for one epoch.
"""

    self.netg.train()
    epoch_iter = 0
    for data in tqdm(self.dataloader['train'], leave=False, total=len(self.dataloader['train'])):
        self.total_steps += self.opt.batchsize
        epoch_iter += self.opt.batchsize

the data is shown ([64,3,128,128]),for the gray dataset .is there another something to modify?

[Coderxiaobai1]

我没有碰到类似的问题 你再看看代码

[lzzlxxlsz]

你用的是黑白图像吗，直接将数据集名称改了，运行就可以了吗

[Coderxiaobai1]

我的是彩色图像。不过我不太懂你的意思，代码写的很好，所以对黑白、彩色图像都支持，要是疑惑的话主要看一下options那个文件，或者运行出来的opt.txt文件，里面具体就是你运行自己数据时的配置。如果还有问题，私聊

[lzzlxxlsz]

我的是彩色图像。不过我不太懂你的意思，代码写的很好，所以对黑白、彩色图像都支持，要是疑惑的话主要看一下options那个文件，或者运行出来的opt.txt文件，里面具体就是你运行自己数据时的配置。如果还有问题，私聊

你好，可以加一下微信吗。18731186906