s6.app.cog.dataloader

Visualize augmented keypoint patches with a DataLoader.

Creates an AugmentedKeypointDataset, wraps it in a torch.utils.data.DataLoader, pulls one batch, and renders the augmented patch with its keypoint overlaid. Useful for sanity‑checking dataset configuration and augmentations.

Examples

Preview a batch of size 8 from a StructuredDataset directory:

>>> python -m s6.app.cog.dataloader ./data/my_dataset ...   --datakeys B.image B.tip_point --batch-size 8 --output-size 256

s6.app.cog.dataloader.main()

Parse arguments, build dataset/loader, and display one batch.