我发现往vutils.make_grid()里面传单通道的黑白图像时,它最后会变成三通道的图像
截取的代码是这样的,其中fake形状是64 * 1 * 64 * 64,是64个黑白图像的数据,而调试的时候往img_list加了一个元素后,元素形状是3 * 64 * 64
我发现往vutils.make_grid()里面传单通道的黑白图像时,它最后会变成三通道的图像
截取的代码是这样的,其中fake形状是64 * 1 * 64 * 64,是64个黑白图像的数据,而调试的时候往img_list加了一个元素后,元素形状是3 * 64 * 64