Crate cifar_ten

Parses the binary files of the CIFAR-10 data set and returns them as a pair of tuples (data, labels) with of type and dimension:

• Training data: Array4<u8> [50_000, 3, 32, 32] and Array2<u8> [50_000, 10]
• Testing data: Array4<u8> [10_000, 3, 32, 32] and Array2<u8> [10_000, 10]

OR

• as a set of flattened Array2<f32> structures in the same arrangement.

A random image from each dataset and the associated label can be displayed upon parsing. A tar.gz file with the original binaries can be found here.

use cifar_ten::*;

fn main() {
    let (train_data, train_labels, test_data, test_labels) = Cifar10::default()
        .show_images(true)
        .build()
        // or .build_as_flat_f32()
        .expect("Failed to build CIFAR-10 data");
}

Dependencies

The crate's show feature uses the minifb library to display sample images, which means you may need to add its dependencies via

sudo apt install libxkbcommon-dev libwayland-cursor0 libwayland-dev

Structs

Cifar10

Data structure used to specify where/how the CIFAR-10 binary data is parsed