Pytorch torchvision.utils.make_grid()用法

时间:2024-03-22 16:58:19

Pytorch torchvision.utils.make_grid()用法

make_grid的作用是将若干幅图像拼成一幅图像。其中padding的作用就是子图像与子图像之间的pad有多宽。nrow是一行放入八个图片。

Pytorch torchvision.utils.make_grid()用法

                 这是padding为0的结果

Pytorch torchvision.utils.make_grid()用法

                    这是padding为2的结果