Crate axonml_vision 

Axonml Vision - Computer Vision Utilities

This crate provides computer vision functionality for the Axonml ML framework:

• Transforms: Image-specific data augmentation and preprocessing
• Datasets: Loaders for common vision datasets (MNIST, CIFAR)
• Models: Pre-defined neural network architectures (LeNet, MLP)

Example

use axonml_vision::prelude::*;

// Load synthetic MNIST data
let train_data = SyntheticMNIST::train();
let test_data = SyntheticMNIST::test();

// Create a LeNet model
let model = LeNet::new();

// Apply image transforms
let transform = Compose::empty()
    .add(ImageNormalize::mnist())
    .add(RandomHorizontalFlip::new());

@version 0.1.0 @author AutomataNexus Development Team

Re-exports

pub use transforms::CenterCrop;

pub use transforms::ColorJitter;

pub use transforms::Grayscale;

pub use transforms::ImageNormalize;

pub use transforms::Pad;

pub use transforms::RandomHorizontalFlip;

pub use transforms::RandomRotation;

pub use transforms::RandomVerticalFlip;

pub use transforms::Resize;

pub use transforms::ToTensorImage;

pub use datasets::FashionMNIST;

pub use datasets::SyntheticCIFAR;

pub use datasets::SyntheticMNIST;

pub use datasets::CIFAR10;

pub use datasets::CIFAR100;

pub use datasets::MNIST;

pub use models::LeNet;

pub use models::SimpleCNN;

pub use models::MLP;

pub use hub::cache_dir;

pub use hub::download_weights;

pub use hub::is_cached;

pub use hub::list_models;

pub use hub::load_state_dict;

pub use hub::model_info;

pub use hub::model_registry;

pub use hub::HubError;

pub use hub::HubResult;

pub use hub::PretrainedModel;

pub use hub::StateDict;

Modules

datasets

Vision Datasets

hub

Model Hub - Pretrained Weights Management

models

Vision Models

prelude

Common imports for computer vision tasks.

transforms

Image Transforms - Vision-Specific Data Augmentation