kaffe 0.2.0

Pytorch, but make it rust :)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//! Module for defining and implementing datasets
#![warn(missing_docs)]

use crate::nn::transform::Transform;
use crate::{Tensor, TensorElement};
use std::error::Error;
use std::str::FromStr;

/// Dataset represents a set of many matrices
pub trait DataSet {
    /// Initializes a new dataset
    fn new(
        img_labels: Vec<String>,
        img_dir: &'static str,
        transform: Option<Transform>,
        target_transform: Option<Transform>,
    ) -> Self;

    /// Gives the size of the dataset
    fn size(&self) -> usize;

    /// Gets an element at a certain index
    fn get<'a, T>(&self, idx: usize) -> (Tensor<'a, T>, String)
    where
        T: TensorElement,
        <T as FromStr>::Err: Error;
}

/// Represents a custom dataset
/// All common datasets will be added here by default
pub struct CustomImageDataset {
    img_labels: Vec<String>,
    img_dir: &'static str,
    transform: Option<Transform>,
    target_transform: Option<Transform>,
}

impl DataSet for CustomImageDataset {
    fn new(
        img_labels: Vec<String>,
        img_dir: &'static str,
        transform: Option<Transform>,
        target_transform: Option<Transform>,
    ) -> Self {
        Self {
            img_labels,
            img_dir,
            transform,
            target_transform,
        }
    }

    fn size(&self) -> usize {
        self.img_labels.len()
    }

    fn get<'a, T>(&self, idx: usize) -> (Tensor<'a, T>, String)
    where
        T: TensorElement,
        <T as FromStr>::Err: Error,
    {
        todo!()
        // img_path = os.path.join(self.img_dir, self.img_labels.iloc[idx, 0])
        // image = read_image(img_path)
        // label = self.img_labels.iloc[idx, 1]
        // if self.transform:
        //     image = self.transform(image)
        // if self.target_transform:
        //     label = self.target_transform(label)
        // return image, label
    }
}