CaveDataset

class colibri.data.sota_datasets.CaveDataset(path, download=True, url='https://www.cs.columbia.edu/CAVE/databases/multispectral/zip/complete_ms_data.zip')

Bases: object

Class to handle the CAVE dataset.

The CAVE dataset is a database of multispectral images that were used to emulate the GAP camera. The images are of a wide variety of real-world materials and objects.

URL: https://www.cs.columbia.edu/CAVE/databases/multispectral/

Parameters:

• path (str)

• download (bool)

• url (str)

download()

Downloads the dataset from the specified URL and extracts it to the specified path.

get_list_paths()

Returns a list of cave filenames in the given path.

Parameters:

path (str) – The path to the directory containing the cave files.

Returns:

A list of cave filenames.

Return type:

list

load_item(filename)

Load a sample from the CAVE dataset. :param filename: The filename of the sample. :type filename: str

Returns:

A dictionary containing the input and output data of the sample.

Return type:

dict

Parameters:

filename (str)