torchvision.utils的解析

torchvision.utils
save_image
make_grid

torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)

把圖片排列成網格形狀。
參數:
tensor(Tensor 或 list)– 四維批(batch)Tensor或列表。如果是Tensor,其形狀應是(B x C x H x W);如果是列表,元素應爲相同大小的圖片。
nrow(int, 可選)– 最終展示的圖片網格中每行擺放的圖片數量。網格的長寬應該是(B / nrow, nrow)。默認是8。
padding(int, 可選)– 擴展填充的像素寬度。默認是2。
normalize(bool, 可選)– 如果設置爲True,通過減去最小像素值然後除以最大像素值,把圖片移到(0,1)的範圍內。
range(tuple, 可選)– 元組(min, max),min和max用於對圖片進行標準化處理。默認的,min和max由輸入的張量計算得到。
scale_each(bool, 可選)– 如果設置爲True,將批中的每張圖片按照各自的最值分別縮放,否則使用當前批中所有圖片的最值(min, max)進行統一縮放。
pad_value(float, 可選)– 擴展填充的像素值。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章