vutils.make_grid()与黑白图像有关的一个小体会

我发现往vutils.make_grid()里面传单通道的黑白图像时,它最后会变成三通道的图像
vutils.make_grid()与黑白图像有关的一个小体会_第1张图片

截取的代码是这样的,其中fake形状是64 * 1 * 64 * 64,是64个黑白图像的数据,而调试的时候往img_list加了一个元素后,元素形状是3 * 64 * 64

你可能感兴趣的:(vutils.make_grid()与黑白图像有关的一个小体会)