torchdatasetutil 0.0.42

Utilities to load and use pytorch datasets stored in Minio S3

Torch Dataset Utilities

The python library torchdatasetutils produces torch DataLoader classes and utility functions for several imaging datasets. This currently includes sets of images and annotations from CVAT, COCO dataset. "torchdatasetutil" uses an s3 object storage to hold dataset data. This enables training and test to be performed on nodes different from where the dataset is stored with application defined credentials. It uses torch PyTorch worker threads to prefetch data for efficient GPU or CPU training and inference.

"torchdatasetutils" takes as an input the pymlutil.s3 object to access the object storage.

Two json or yaml dictionaries are loaded from the object storage to identify and process the dataset: the dataset description and class dictionary. The the dataset description is unique for each type of dataset. The class dictionary is common to all datasets and describes data transformation and data augmentation.

Library structure

• pymlutil.s3: access to object storage
• torchdatasetutil
  • gitcoco.getcoco: function to load the COCO dataset from internet archives into object storage
  • cocostore
    • CocoStore: class providing a python iterator over the coco dataset in object storage
    • CocoDataset" class implementing the pytorch Dataset class for the CocoStore iterator
  • imstore

See torchdatasetutil.ipynb for library interface and usage