Open Neural Network Exchange (ONNX) is an open standard format for representing machine learning models. Read more
The DeepHealth Model Zoo is a collection of pre-trained models in the ONNX format, that we have check that they are compatible with the lastest version of the EDDL library.
We have standardized on Git LFS (Large File Storage) to store ONNX model files. To download an ONNX model, navigate to the appropriate Github page and click the Download button on the top right.
Disclaimer:
This document is a modified version of onnx/models repository.
This collection of models take images as input, then classifies the major objects in the images into 1000 object categories such as keyboard, mouse, pencil, and many animals.
| Model Class | Reference | Description |
|---|---|---|
| ResNet | He et al. | A CNN model (up to 152 layers). Uses shortcut connections to achieve higher accuracy when classifying images. Top-5 error from paper - ~3.6% |
Object detection models detect the presence of multiple objects in an image and segment out areas of the image where the objects are detected. Semantic segmentation models partition an input image by labeling each pixel into a set of pre-defined categories.
| Model Class | Reference | Description |
|---|