视觉

此库提供对常用机器学习基准数据集的访问。

该库目前包括

• MNIST
• FashionMNIST
• CIFAR-10
• CIFAR-100

目前功能非常基础。

使用方法

每个数据集都可以使用 Builder 类下载和处理。每个情况下Builder都是可定制的。

extern crate vision;

use vision::mnist::{MNISTBuilder};

fn main() {
    let builder = MNISTBuilder::new();
    let mnist = builder.data_home("MNIST")
                       .verbose()
                       .get_data().unwrap();
    println!("{}", mnist.train_imgs.len());
}

Builder 返回的 MNIST 对象包含四个公共字段，train_imgs、train_labels、test_images 和 test_labels。标签字段是 Vec<u8> 类型，图像是 Vec<Vec<u8>> 类型，外层 Vec 中的每个条目对应一个数据点。

进一步的预处理应由用户执行。