1.4 KiB
1.4 KiB
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% |