Script to create the GIF shown in https://twitter.com/TorchIOLib/status/1459841764576141315

medmnist_gif.py

import torch
import torchio as tio
from einops import rearrange

rows = 16
cols = 28

dataset = tio.datasets.OrganMNIST3D('train')
batch_size = rows * cols
loader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=True)
batch = tio.utils.get_first_item(loader)
tensor = batch['image'][tio.DATA]
pattern = '(b1 b2) c x y z -> c x (b1 y) (b2 z)'
tensor = rearrange(tensor, pattern, b1=cols, b2=rows)
image = tio.ScalarImage(tensor=tensor)
image.to_gif(0, 2, f'/tmp/{dataset.__class__.__name__}.gif')