Rust – MNIST データベースをダウンロードして解凍する

[package]
name = "deep-learning-rs"
version = "0.1.0"
edition = "2021"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
reqwest = "*"
tokio = { version = "*", features = ["full"] }
flate2 = "*"

use flate2::read::GzDecoder;
use std::fs::File;
use std::io::copy;

#[tokio::main]
async fn main() {
    let base_url = "http://yann.lecun.com/exdb/mnist/";
    let filenames = [
        "train-images-idx3-ubyte.gz",
        "train-labels-idx1-ubyte.gz",
        "t10k-images-idx3-ubyte.gz",
        "t10k-labels-idx1-ubyte.gz",
    ];

    for filename in filenames {
        let url = format!("{}{}", base_url, filename);
        let content = reqwest::get(url).await.unwrap().bytes().await.unwrap();

        let mut gz = GzDecoder::new(content.as_ref());
        let dst_filename = filename.split('.').next().unwrap();
        let mut dst = File::create(dst_filename).unwrap();
        copy(&mut gz, &mut dst).unwrap();
    }
}