寻找 torch.imshow() 'like' 命令
Looking for a torch.imshow() 'like' command
假设我有一个可变图像（当前位于 gpu 上），大小为 [32,1,256,256]，其中 32 是批量大小，1 是通道数量（灰度）。
而不是绘制这个：
plt.imshow(img[0,0,:,:].cpu().detach(),'gray');plt.show()
我希望我能做到
torch.imshow(img,8,'gray') 它将从我的批次中绘制 8 张图像
有这样的吗？
您正在寻找 torchvision.utils.make_grid：它将 [32, 1, 256,256] 张量转换为 32 个图像的网格。您仍然需要使用 plt 将图像网格实际绘制到屏幕上。
假设我有一个可变图像（当前位于 gpu 上），大小为 [32,1,256,256]，其中 32 是批量大小，1 是通道数量（灰度）。
而不是绘制这个：
plt.imshow(img[0,0,:,:].cpu().detach(),'gray');plt.show()
我希望我能做到
torch.imshow(img,8,'gray') 它将从我的批次中绘制 8 张图像
有这样的吗？
您正在寻找 torchvision.utils.make_grid：它将 [32, 1, 256,256] 张量转换为 32 个图像的网格。您仍然需要使用 plt 将图像网格实际绘制到屏幕上。