Oekk Model Zoo

A collection of pre-trained, state-of-the-art models in different formats with scripts to generate them and converter tools.

Models

Vision

• CNN (Convolutional Neural Networks)
  • Object Detection & Image Segmentation
  • MNIST models

CNN

Model Class	Reference	Description
U-Net	Olaf Ronneberger et al.	CNN with very good performance on very different biomedical segmentation applications.

Object Detection & Image Segmentation

MNIST models

This collection of models take 32x32x1 images (or variations) as input, then classifies the images into 10 number categories. Solvers of the MNIST dataset, frequently used as guinea pigs.

Model Class	Reference	Description
LeNet	Yann LeCun et al.	LeNet is one of the first CNN for image recognition.
SqueezeNet	Iandola et al.	A light-weight CNN model providing AlexNet level accuracy with 50x fewer parameters. Top-5 error from paper - ~20%