Skip to content

Regrouping all neural networks for Kalray Neural Networks applications

License

Notifications You must be signed in to change notification settings

kalray/kann-models-zoo

Repository files navigation

Kalray Neural Network Models


ACE5.3.0 ACE5.3.1 Classifiers Object-Detect Segmentation

The KaNN™ Model Zoo repository provides a list of neural networks models ready to compile & run on MPPA® manycore processor. This comes on top of KaNN™ tool for model generation and enhance AI solutions onto Kalray processor.

SDK Kalray Neural Network (KaNN)

Kalray Neural Network (KaNN) is a SDK included in AccessCore Embedded (ACE) compute offer to optimize AI inference on our dedicated processor called MPPA® (last generation, the 3rd, is named Coolidge 2). It is composed by:

  • generator : a python wheel to parse, optimize and paralellize an intermediate representation of a neural networks. Thanks to the runtime, it gives you then the opportunity to run the algorithm directly on the MPPA®
  • runtime : optimized libraries (in ASM/C/C++) to execute each operation nodes.

ACE-5.3.0 / ACE-5.3.1 (KaNN) supports: Tensorflow, TFlite, ONNX and Pytorch/ONNX.

Contents

To quickly deploy a neural network on the MPPA®, a WIKI note is available here:

CNN Models are divided into 3 types of Machine Vision applications:

The examples below illustrates the kind of predictions you must have:

Classification (Regnet-x-1.6g) Object-detection (Yolov8n) Segmentation (Deeplabv3+)

*images has been realized with the model of this repository and KaNN™ SDK solution

List of Neural Networks

All networks are proposed into selected Neural Network architectures, such as:

Classifiers : complete list can be found here

  • DenseNet
  • EfficientNet
  • Inception
  • ResNet
  • RegNet
  • MobileNet
  • NasNet
  • SqueezeNet
  • VGG

Object-detection : complete list can be found here

  • EfficientDet
  • Faster-RCNN
  • FCN
  • RetinatNet
  • SSD
  • YOLO

Segmentation : complete list can be found here

  • DeeplabV3+
  • Mask-RCNN
  • UNet
  • YOLO

Requirements

Hardware requirements

Host machine(s):

  • x86_64 CPU
  • DDR RAM 8Go min
  • HDD disk 32 Go min
  • PCIe Gen3 min, Gen4 recommended

Acceleration card(s):

  • A
  • A

Software requirements

  • U22 Ker
  • ACE
  • ACE
  • Python Python

About

Regrouping all neural networks for Kalray Neural Networks applications

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages