PyTorch
Among academic researchers, PyTorch is the most popular Python deep learning framework.
You can use PyTorch to train deep learning models while using SwanLab for experiment tracking and visualization.
Log Tensor Images
You can pass PyTorch Tensors with image data to swanlab.Image, and swanlab.Image will use torchvision to convert them into images:
python
image_tensors = ... # Tensor images with shape [B, C, H, W]
swanlab.log({"examples": [swanlab.Image(im) for im in image_tensors]})