Code Monkey home page Code Monkey logo

ml-zoo's Introduction

Model Zoo

version

A collection of machine learning models optimized for Arm IP.

Anomaly Detection

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (AUC)
MicroNet Large INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.968
MicroNet Medium INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.963
MicroNet Small INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.955

Dataset: Dcase 2020 Task 2 Slide Rail

Image Classification

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (Top 1 Accuracy)
MobileNet v2 1.0 224 INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.697
MobileNet v2 1.0 224 UINT8 UINT8 TensorFlow Lite ✖️ ✖️ ✔️ ✔️ 0.708

Dataset: ILSVRC 2012

Keyword Spotting

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (Accuracy)
CNN Large INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.931
CNN Medium INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.911
CNN Small INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.912
DNN Large INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.863
DNN Medium INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.844
DNN Small INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.825
DS-CNN Clustered FP32 * FP32 TensorFlow Lite ✔️ ✖️ ✔️ ✖️ 0.950
DS-CNN Clustered INT8 * INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.940
DS-CNN Large INT8 * INT8 TensorFlow Lite ✔️ ✔️ HERO ✔️ ✔️ 0.946
DS-CNN Medium INT8 * INT8 TensorFlow Lite ✔️ ✔️ HERO ✔️ ✔️ 0.941
DS-CNN Small INT8 * INT8 TensorFlow Lite ✔️ ✔️ HERO ✔️ ✔️ 0.935
MicroNet Large INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.965
MicroNet Medium INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.958
MicroNet Small INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.953

Dataset: Google Speech Commands Test Set

Noise Suppression

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (Average Pesq)
RNNoise INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 2.945

Dataset: Noisy Speech Database For Training Speech Enhancement Algorithms And Tts Models

Object Detection

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (mAP)
SSD MobileNet v1 FP32 * FP32 TensorFlow Lite ✔️ ✖️ ✔️ ✖️ 0.210
SSD MobileNet v1 INT8 * INT8 TensorFlow Lite ✔️ ✖️ ✔️ ✖️ 0.234
SSD MobileNet v1 UINT8 * UINT8 TensorFlow Lite ✖️ ✖️ ✔️ ✖️ 0.180
YOLO v3 Tiny FP32 * FP32 TensorFlow Lite ✔️ ✖️ ✔️ ✖️ 0.331

Dataset: COCO Validation 2017

Speech Recognition

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (LER)
Wav2letter INT8 INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.0877
Wav2letter Pruned INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✔️ ✔️ 0.0783
Tiny Wav2letter INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✖️ ✔️ 0.0348
Tiny Wav2letter Pruned INT8 * INT8 TensorFlow Lite ✔️ ✔️ ✖️ ✔️ 0.0283

Dataset: LibriSpeech, Fluent Speech

Superresolution

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (PSNR)
SESR INT8 ** INT8 TensorFlow Lite ✔️ ✖️ ✔️ HERO ✖️ 35.00dB

Dataset: DIV2K

Visual Wake Words

Network Type Framework Cortex-A Cortex-M Mali GPU Ethos U Score (Accuracy)
MicroNet VWW-2 INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.768
MicroNet VWW-3 INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.855
MicroNet VWW-4 INT8 INT8 TensorFlow Lite ✖️ ✔️ ✔️ ✔️ 0.822

Dataset: Visual Wake Words

Key

  • ✔️ - Will run on this platform.
  • ✖️ - Will not run on this platform.
  • * - Code to recreate model available.
  • ** - This model has a large memory footprint – it will not run on all platforms.

License

Apache-2.0 unless otherwise explicitly stated.

ml-zoo's People

Contributors

tom-arm avatar orly-arm avatar idan-arm avatar chenhayat avatar gregory-bronstain avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.