假設我有一個可變影像(當前位于 gpu 上),大小為[32,1,256,256]32 是批量大小,1 是通道數量(灰度)。
而不是繪制這個:
plt.imshow(img[0,0,:,:].cpu().detach(),'gray');plt.show()
我希望我能做到
torch.imshow(img,8,'gray'),它會從我的批次中繪制 8 張影像,有這樣的事情嗎?
uj5u.com熱心網友回復:
您正在尋找torchvision.utils.make_grid:它將[32, 1, 256,256]張量轉換為 32 個影像的網格。您仍然需要使用plt實際將影像網格繪制到螢屏上。
轉載請註明出處,本文鏈接:https://www.uj5u.com/qiye/407668.html
標籤:
