tch_plus/vision/
mod.rs

1//! The `vision` module groups functions and models related to
2//! computer vision.
3pub mod dataset;
4
5pub mod image;
6
7pub mod mnist;
8
9pub mod cifar;
10
11pub mod alexnet;
12
13pub mod inception;
14
15pub mod resnet;
16
17pub mod densenet;
18
19pub mod vgg;
20
21pub mod squeezenet;
22
23pub mod mobilenet;
24
25pub mod imagenet;
26
27pub mod efficientnet;
28
29pub mod convmixer;
30
31pub mod dinov2;
32
33#[cfg(feature = "image")]
34mod rust_image;