Skip to content

image-classification

Directory actions

More options

Directory actions

More options

Latest commit

 

History

History

image-classification

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
beit.rs
beit.rs
 
 
convnext.rs
convnext.rs
 
 
deit.rs
deit.rs
 
 
fastvit.rs
fastvit.rs
 
 
main.rs
main.rs
 
 
mobileone.rs
mobileone.rs
 
 
ram.rs
ram.rs
 
 

README.md

Image Classification Examples

This directory contains examples for image classification using various SOTA models.

Usage

# BeiT example
cargo run -F cuda-full -F vlm --example image-classification -- beit    --device cuda --processor-device cuda 

# DeiT example
cargo run -F cuda-full -F vlm --example image-classification -- deit    --device cuda --processor-device cuda 

# MobileOne example
cargo run -F cuda-full -F vlm --example image-classification -- mobileone    --device cuda --processor-device cuda --kind s0

# ConvNeXt example
cargo run -F cuda-full -F vlm --example image-classification -- convnext    --device cuda --processor-device cuda

# FastViT
cargo run -F cuda-full -F vlm --example image-classification -- fastvit    --device cuda --processor-device cuda

# RAM example
cargo run -F cuda-full -F vlm --example image-classification -- ram    --device cuda --processor-device cuda

# RAM++ example
cargo run -F cuda-full -F vlm --example image-classification -- ram    --device cuda --processor-device cuda  --variant ram++