11 lines
523 B
Python
11 lines
523 B
Python
import torchvision.utils
|
|
|
|
|
|
def make_2d_grid(tensors, padding=0, normalize=True, range=None, scale_each=False, pad_value=0):
|
|
# merge image in a batch in `y` direction first.
|
|
grids = [torchvision.utils.make_grid(img_batch, padding=padding, nrow=1, normalize=normalize, range=range,
|
|
scale_each=scale_each, pad_value=pad_value)
|
|
for img_batch in tensors]
|
|
# merge images in `x` direction.
|
|
return torchvision.utils.make_grid(grids, padding=0, nrow=len(grids))
|