onnxruntime_ng/download/vision/
domain_based_image_classification.rs

1//! Module defining domain-based image classification models available to download.
2//!
3//! See [https://github.com/onnx/models#domain-based-image-classification-](https://github.com/onnx/models#domain-based-image-classification-)
4
5use crate::download::{vision::Vision, AvailableOnnxModel, ModelUrl};
6
7/// Image classification model
8#[derive(Debug, Clone)]
9pub enum DomainBasedImageClassification {
10    /// Handwritten digits prediction using CNN
11    ///
12    /// Source: [https://github.com/onnx/models/tree/master/vision/classification/mnist](https://github.com/onnx/models/tree/master/vision/classification/mnist)
13    ///
14    /// Variant downloaded: ONNX Version 1.3 with Opset Version 8.
15    Mnist,
16}
17
18impl ModelUrl for DomainBasedImageClassification {
19    fn fetch_url(&self) -> &'static str {
20        match self {
21            DomainBasedImageClassification::Mnist => "https://github.com/onnx/models/raw/master/vision/classification/mnist/model/mnist-8.onnx",
22        }
23    }
24}
25
26impl From<DomainBasedImageClassification> for AvailableOnnxModel {
27    fn from(model: DomainBasedImageClassification) -> Self {
28        AvailableOnnxModel::Vision(Vision::DomainBasedImageClassification(model))
29    }
30}