Metadata-Version: 2.4
Name: modelmaite
Version: 0.1.0
Summary: MAITE-compliant model wrappers built on the maite protocols
License: Apache-2.0
License-File: LICENSE.md
Keywords: jatic,artificial intelligence,maite,models
Author: OpenTeams
Requires-Python: >=3.10, <3.13
Classifier: Development Status :: 3 - Alpha
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Provides-Extra: onnx
Provides-Extra: onnx-cuda
Provides-Extra: torchvision
Provides-Extra: visdrone
Requires-Dist: httpx (>=0.28.1) ; extra == "visdrone"
Requires-Dist: numpy (>=1.24,<2)
Requires-Dist: onnx (>=1.14.0) ; extra == "onnx"
Requires-Dist: onnx (>=1.14.0) ; extra == "onnx-cuda"
Requires-Dist: onnxruntime (>=1.14.0,<1.24.0) ; extra == "onnx"
Requires-Dist: onnxruntime-gpu (>=1.14.0,<1.24.0) ; extra == "onnx-cuda"
Requires-Dist: smqtk-detection (>=0.22.0) ; extra == "visdrone"
Requires-Dist: torch (>=2.2.0) ; extra == "torchvision"
Requires-Dist: torch (>=2.2.0) ; extra == "visdrone"
Requires-Dist: torchvision (>=0.17.0) ; extra == "torchvision"
Requires-Dist: typing-extensions (>=4.5)
Description-Content-Type: text/markdown

# modelmaite - MAITE-compliant model wrappers

`modelmaite` is a Python package providing model utilities built on the
[maite](https://mit-ll-ai-technology.github.io/maite/) protocols.

**THIS PACKAGE IS CURRENTLY UNDER CONSTRUCTION**

## Model wrappers

`modelmaite.image_classification.TorchvisionICModel` wraps torchvision image-classification
models as MAITE-compatible image-classification models.

`modelmaite.image_classification.OnnxICModel` wraps JATIC_ONNX v1 image-classification
models as MAITE-compatible image-classification models.

`modelmaite.object_detection.TorchvisionODModel` wraps torchvision object-detection
models as MAITE-compatible object-detection models.

`modelmaite.object_detection.VisdroneODModel` wraps Kitware CenterNet VisDrone
models as MAITE-compatible object-detection models.

`modelmaite.object_detection.OnnxODModel` wraps JATIC_ONNX v1 object-detection
models as MAITE-compatible object-detection models.

Install the optional torchvision dependencies with Poetry:

```bash
poetry add "modelmaite[torchvision]"
```

Install the optional VisDrone dependencies with Poetry:

```bash
poetry add "modelmaite[visdrone]"
```

Install the optional ONNX Runtime dependencies with Poetry:

```bash
poetry add "modelmaite[onnx]"
```

