Skip to content

Repository files navigation

Akida Model Zoo

Introduction

The Akida Model Zoo expands our foundation models with a hand-picked collection of models accelerated by the Akida 2.0 IP. Designed for developers, researchers, and AI enthusiasts, these ready-to-use models make it easier than ever to explore, build, and innovate with the Akida solution.

Models

Both float and quantized models are available, with quantized versions converted and evaluated on the Akida solution. For each model, the number of nodes required to run on a minimal Akida IP configuration is provided, enabling straightforward assessment of performance and deployment needs.

In addition, some models can be evaluated directly through Akida Cloud ☁️, offering a convenient way to explore and experiment without local hardware.

DomainUse caseArchitectureResolutionDataset#ParamsQuantizationAccuracyF1 ScoreMSEMinimal #Nodes
VisionClassificationMobileNetV1_1.0128CIFAR-102.25M891.92%5 ☁️
VisionClassificationMobileNetV1_1.0224Oxford_Flower3.3M891.08%7
VisionClassificationMobileNetV1 0.5224SIIM-ISIC3.14M898.16%86.14%4 ☁️
VisionClassificationMobileNetV1 0.5224ODIR-5K3.15M892.00%97.83%4 ☁️
VisionClassificationMobileNetV1 0.5224ECG3.14M883.27%89.18%4 ☁️
VisionClassificationMobileNetV1 0.5224Retina OCT3.15M893.30%98.66%4 ☁️
VisionClassificationMobileNetV2 1.0224ImageNet3.5M870.35%7
VisionClassificationMobileNetV2 0.75160ImageNet2.6M862.85%4 ☁️
VisionClassificationMobileNetV2 0.3596ImageNet1.2M843.47%2 ☁️
VisionClassificationMobileNetV4 1.0224ImageNet3.77M871.86%8
VisionClassificationMobileNetV2_1.0128CIFAR-102.25M893.96%5 ☁️
VisionClassificationMobileNetV2_1.0224Oxford_Flower2.4M891.97%7
VisionClassificationMobileNetV4_1.0128CIFAR-102.5M894.72%7
VisionClassificationMobileNetV4_1.0224Oxford_Flower2.6M885.41%8
VisionClassificationspatiotemporal224FallVision1.34M898.36%16
VisionClassificationMLP784MNIST203.5K898.05%1 ☁️
VisionClassificationLogisticRegression784MNIST7.9K884.52%1 ☁️
ECGClassification1DCNN360MIT-BIH74K897.3%1 ☁️
ECGAnomaly Detection1DCNN144ECG5000290K894.0%1 ☁️
TabularClassificationLogisticR.30Breast_Cancer169893.9%1 ☁️
SyntheticRegressionMLP11D_Curve6.2K80.1361 ☁️
VisionDetectionAkidaNet18/CenterNet224Soda_bottle2.43M891.53%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Trail_camera2.43M884.74%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Road_signs2.43M865.46%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Furniture2.43M879.21%6 ☁️
VisionDetectionAkidaNet18/CenterNet384Aerial_Cows2.43M830.91%16
VisionDetectionAkidaNet18/CenterNet224Bees2.43M859.99%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Cable_Damage2.43M877.32%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Corrosion2.43M839.06%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Digits2.43M892.32%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Face_Detection2.43M875.98%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Hand_Gestures2.43M853.52%6 ☁️
VisionDetectionAkidaNet18/CenterNet224License_Plate2.43M896.22%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Liver_Disease2.43M840.57%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Phages2.43M867.18%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Sign_Language2.43M885.88%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Stomata_Cells2.43M852.14%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Underwater_Objects2.43M844.89%6 ☁️
VisionDetectionAkidaNet18/CenterNet384Ships_Detection2.43M839.60%12
VisionDetectionAkidaNet18/CenterNet224Bone_Fracture2.43M860.70%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Facial_Expression2.43M875.40%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Pothole_Detection2.43M857.20%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Search_And_Rescue2.43M877.00%6 ☁️
VisionDetectionAkidaNet18/CenterNet224Traffic_Detection2.43M871.80%6 ☁️
VisionDetectionAkidaNet18/CenterNet384Weed_Crop2.43M847.70%16

Download

Git Clone

To avoid downloading the models during cloning due to their large size:

GIT_LFS_SKIP_SMUDGE=1 git clone git@github.com:Brainchip-Inc/models.git

To download a specific model:

git lfs pull --include="[path to model]" --exclude=""

To download all models:

git lfs pull --include="*" --exclude=""

GitHub UI

Alternatively, you can download models directly from GitHub. Navigate to the model's page and click the "Download" button on the top right corner.

Model Visualization

For a graphical representation of each model's architecture, we recommend using Netron.

About

An extension of Akida model zoo with pre-trained models

Resources

Stars

4 stars

Watchers

3 watching

Forks

Releases

Contributors

Languages