PyTorchDataset

ApertureDBDataset

ApertureDBDataset(client: Connector, query, label_prop=None, batch_size=1, command_idx=None)

Bases: Dataset

This class implements a PyTorch Dataset for ApertureDB. It is used to load blobs returned by a Find* command from ApertureDB into a PyTorch model. It can be initialized with a query that will be used to retrieve the blobs from ApertureDB. Note that only FindImage blobs are decoded via OpenCV.