ovkit 0.2.0__tar.gz → 0.4.0__tar.gz

{ovkit-0.2.0 → ovkit-0.4.0}/.gitignore

@@ -217,3 +217,29 @@ __marimo__/
 
 # Streamlit
 .streamlit/secrets.toml
+
+# --- what running ovkit leaves behind ---------------------------------------
+#   python examples/speak.py        -> speak.wav, speak.png
+#   ovkit run detect photo.jpg      -> photo_out.jpg
+#   python scripts/verify_capabilities.py -> .verify/
+#
+# Examples and scripts write their results into the folder you ran them from,
+# which for a fresh clone is the repo root. Every rule here is rooted (leading
+# slash) on purpose: a picture someone deliberately adds under docs/ or tests/
+# is still tracked, and only the clutter at the top is hidden.
+/*.jpg
+/*.jpeg
+/*.png
+/*.bmp
+/*.webp
+/*.wav
+/*.mp4
+# `ovkit run` names its output after the input, whatever the extension.
+/*_out.*
+# examples/teach_ai.py — a classifier you taught, saved next to the code.
+/my-ai/
+# Scratch folders the scripts default to.
+/.verify/
+/tts_out/
+/supertonic_ir/
+/report.txt

{ovkit-0.2.0 → ovkit-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: ovkit
-Version: 0.2.0
+Version: 0.4.0
 Summary: A simple Python inference API for OpenVINO: one import, one Model class, clean Results — with AUTO/NPU, async, and INT8.
 Project-URL: Homepage, https://github.com/leeyunjai82/ovkit
 Project-URL: Documentation, https://leeyunjai82.github.io/ovkit/
@@ -27,9 +27,12 @@ Requires-Dist: pillow>=9.0
 Requires-Dist: pyyaml>=6.0
 Provides-Extra: all
 Requires-Dist: anomalib; extra == 'all'
+Requires-Dist: mediapipe; extra == 'all'
 Requires-Dist: nncf; extra == 'all'
 Requires-Dist: openvino-genai; extra == 'all'
-Requires-Dist: optimum-intel; extra == 'all'
+Requires-Dist: optimum-intel[openvino]>=1.20; extra == 'all'
+Requires-Dist: rtdetr[train]>=0.6; extra == 'all'
+Requires-Dist: transformers>=4.40; extra == 'all'
 Provides-Extra: anomaly
 Requires-Dist: anomalib; extra == 'anomaly'
 Provides-Extra: dev
@@ -39,8 +42,15 @@ Requires-Dist: ruff; extra == 'dev'
 Provides-Extra: genai
 Requires-Dist: openvino-genai; extra == 'genai'
 Requires-Dist: optimum-intel; extra == 'genai'
+Provides-Extra: hand
+Requires-Dist: mediapipe; extra == 'hand'
+Provides-Extra: hf
+Requires-Dist: optimum-intel[openvino]>=1.20; extra == 'hf'
+Requires-Dist: transformers>=4.40; extra == 'hf'
 Provides-Extra: quant
 Requires-Dist: nncf; extra == 'quant'
+Provides-Extra: train
+Requires-Dist: rtdetr[train]>=0.6; extra == 'train'
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -49,8 +59,9 @@ Description-Content-Type: text/markdown
 
 # ovkit
 
-**OpenVINO inference in 3 lines.** One `Model` class, clean `Results`, 30+
-ready models — with `AUTO`/`NPU`/`GPU` devices, async throughput, and INT8.
+**OpenVINO inference in one line.** One `Model` class, clean `Results`, and 19
+composed capabilities over 68 ready models — with `AUTO`/`NPU`/`GPU` devices,
+async throughput, INT8, and your own detector trainable in the same package.
 
 [![CI](https://github.com/leeyunjai82/ovkit/actions/workflows/ci.yml/badge.svg)](https://github.com/leeyunjai82/ovkit/actions/workflows/ci.yml)
 [![Docs](https://img.shields.io/badge/docs-github.io-0a7d8c)](https://leeyunjai82.github.io/ovkit/)
@@ -83,6 +94,8 @@ for row in r.found:                    # [{'name': '사람', 'name_en': 'person'
 
 for r in Model("track", 0):            # webcam / video / "mic" -> a stream
     print(r, r.elapsed_ms, r.device)   # 2x person (#1, #4) 14.2 GPU
+    if not r.show("track"):            # a window; q or Esc ends the loop
+        break
 ```
 
 The same call also runs whole **capabilities** — several models chained into one
@@ -114,6 +127,8 @@ Python 3.10+. For development, install from source:
 git clone https://github.com/leeyunjai82/ovkit.git && cd ovkit
 python -m venv .venv && source .venv/bin/activate   # Windows: .venv\Scripts\activate
 pip install -e ".[dev]"
+
+python scripts/check.py --fix    # what CI runs: ruff, black, pytest
 ```
 
 ## Capabilities
@@ -138,6 +153,13 @@ camera index).
 | `attention` | `1 person looking at: laptop` | gaze + object detection (ray-cast into the boxes) |
 | `anonymize` | the picture with every face pixelated | face (and plate) detection + redaction |
 | `face_match` | `('yunjai', 0.81)` — who this is | embedding + cosine matching against your gallery |
+| `count` | `pencil 3 · cup 1` (optionally one kind only) | detection + per-kind tally |
+| `posture` | `neck 32° — sit up (6s)` | pose + neck angle **over time** |
+| `exercise` | `squat x 12 (down)` | pose + joint-angle hysteresis (squat, push-up) |
+| `attendance` | `present 24/26` + `roll.csv` | face detection + roster matching |
+| `teach` | your own classes, from example photos | embedding + cosine k-NN (5 modes) |
+| `depth` | `nearest: bottom-left · 34% of the frame is close` + a colour map | Depth Anything V2 Small |
+| `remove_background` | the subject, saved as a transparent PNG | U2-Net |
 
 ```python
 from ovkit import Model, list_pipelines
@@ -149,8 +171,78 @@ Model("face_analyze", attributes=("age_gender",))   # configure what runs
 ```
 
 Aliases: `ocr`, `anpr`, `blur`, `driver`, `describe`, `faces`, `people`, `vehicle`, `tracking`, `reid`.
-`ovkit capabilities` prints the list; **`ovkit gui` opens a window** where you can
-click through them against your webcam or a picture.
+
+## Teach your own AI (no GPU)
+
+A Teachable Machine in five lines — an embedding model turns examples into
+vectors, new inputs match the nearest ones. Five modes: `photo` (default),
+`face` (expressions), `hand` (needs `ovkit[hand]`), `upper`, `body`:
+
+```python
+ai = Model("teach")                       # or Model("가르치기")
+ai.learn("can", "photos/cans/")           # a folder per thing to recognise
+ai.learn("bottle", "photos/bottles/")
+print(ai.guess("new_photo.jpg"))          # ('can', 0.93)
+print(ai.score("test_photos/"))           # accuracy + what it confuses
+ai.save("recycling")                      # -> Documents/ovkit/recycling.json
+for r in ai.predict(0, stream=True): ...  # webcam over YOUR classes
+```
+
+Korean method names work too: `배우기`, `맞혀봐`, `점수`, `저장`. Collect
+examples with the webcam: `from ovkit.pipelines.teach import collect; collect("가위", 30)`.
+
+## Any model on the Hugging Face Hub
+
+The Open Model Zoo is an archive; the models people want are on the Hub.
+`optimum-intel` converts almost any of them to OpenVINO IR, so ovkit does not
+have to mirror a model to serve it:
+
+```bash
+pip install "ovkit[hf]"                      # the converter — needed once
+ovkit pull google/vit-base-patch16-224
+```
+
+```python
+Model("google/vit-base-patch16-224", "photo.jpg")   # tabby 0.94
+```
+
+Converting is the heavy half and happens once; what lands in the cache is a
+plain OpenVINO model, so **running it afterwards needs nothing but ovkit** — a
+classroom machine can be handed the cache and never see PyTorch. The pull also
+writes the model's own class names and the normalisation its image processor
+declares, so results read as answers and not as confident nonsense.
+
+Tasks: image classification, zero-shot (CLIP) classification, feature
+extraction. Detection and segmentation come from the registry instead
+(`detect`, `segment`) — optimum-intel does not export those.
+
+## Train your own detector
+
+RT-DETR, trainable, Apache-2.0 end to end — an Ultralytics-style workflow with
+no AGPL code or weights anywhere (`pip install "ovkit[train]"`):
+
+```python
+from ovkit import RTDETR
+
+model = RTDETR("rtdetr-r18")                 # COCO weights, downloaded on first use
+model.train(data="data.yaml", epochs=100)    # the YOLO-format labels you already have
+model.val(data="data.yaml").box.map50        # mAP50 / mAP50-95
+model.export(half=True)                      # -> IR + labels.txt
+```
+
+The three sizes are `rtdetr_r18` (20M, 46.4 COCO AP — what `detect` uses),
+`rtdetr_r34` (31M, 48.9) and `rtdetr_r50` (43M, 53.1), mirrored like every
+other model. Running them needs nothing but ovkit; `ovkit[train]` adds PyTorch
+for fine-tuning:
+
+```python
+Model("detect", "street.jpg")      # r18 — keeps up with a webcam
+Model("rtdetr_r50", "street.jpg")  # when you want the accuracy instead
+```
+
+A fine-tuned model's exported IR drops straight into `Model("path/to/best.xml")` — the
+`labels.txt` written next to it means your classes answer by name. CLI:
+`ovkit train --data data.yaml`, `ovkit val`, `ovkit export`.
 
 ## Supported tasks
 
@@ -169,7 +261,7 @@ Every single-model task below runs end-to-end through the same 3 lines — swap
 | LLM / STT (GenAI) | `llm`, `stt` | generated text | [llm.py](https://github.com/leeyunjai82/ovkit/blob/main/examples/llm.py) / [stt.py](https://github.com/leeyunjai82/ovkit/blob/main/examples/stt.py) |
 | NLP / audio / time series | `qa`, `translation`, `noise_suppression`, `time_series` | tensors via `model.infer()` | [denoise_audio.py](https://github.com/leeyunjai82/ovkit/blob/main/examples/denoise_audio.py) |
 
-The registry exposes **one well-tested model per capability (35 total)**; the
+The registry exposes **one well-tested model per capability (43 OpenVINO IR + 13 GenAI)**; the
 [HF mirror](https://huggingface.co/leeyunjai/ovkit-models) hosts the full
 Apache-2.0 OMZ set (other tiers, `int8`, `sparse` variants) — surfacing a
 variant is a one-line edit ([catalog](https://leeyunjai82.github.io/ovkit/models.html)).
@@ -261,7 +353,7 @@ ovkit devices                               # available OpenVINO devices
 ```python
 from ovkit.genai import pipeline
 
-llm = pipeline("llm")                        # tinyllama_chat from the mirror
+llm = pipeline("llm")                        # qwen25_1_5b_instruct from the mirror
 print(llm.generate("Explain OpenVINO in one sentence.", max_new_tokens=64))
 
 stt = pipeline("stt")                        # whisper_base
@@ -285,6 +377,40 @@ automatically and results render with overlays.
 
 </details>
 
+## The model mirror
+
+Everything ovkit downloads comes from one repository,
+[leeyunjai/ovkit-models](https://huggingface.co/leeyunjai/ovkit-models) — one
+copy for an offline site to take, one repository to keep alive. Models that
+originate elsewhere (the RT-DETR sizes, Depth Anything V2 Small, U2-Net) are
+copied in, and their home stays registered as the fallback.
+
+To refresh it after adding a model, run the **Sync the model mirror** workflow
+from the Actions tab (dry run by default; tick *upload* to copy). It needs an
+`HF_TOKEN` repository secret with write access, set once. The same thing runs
+locally:
+
+```bash
+export HF_TOKEN=hf_...
+python scripts/sync_mirror.py            # what is missing
+python scripts/sync_mirror.py --upload   # copy just that
+```
+
+It lists the target first and copies only what is absent, so re-running is safe.
+
+The traffic goes the other way too. A mirror only ever added to fills up with
+models that left the lineup and weights uploaded under an old path — and it is
+the thing a school clones whole:
+
+```bash
+python scripts/audit_mirror.py           # what no manifest references
+python scripts/audit_mirror.py --prune   # delete just that
+```
+
+A model's `README.md`, `LICENSE` and `labels.txt` stay as long as the model
+does, and go with it when it leaves. Deletions are ordinary Hub commits, so
+the history still has them.
+
 ## Adding a model
 
 Models are data, not code — one manifest entry

{ovkit-0.2.0 → ovkit-0.4.0}/README.md

@@ -4,8 +4,9 @@
 
 # ovkit
 
-**OpenVINO inference in 3 lines.** One `Model` class, clean `Results`, 30+
-ready models — with `AUTO`/`NPU`/`GPU` devices, async throughput, and INT8.
+**OpenVINO inference in one line.** One `Model` class, clean `Results`, and 19
+composed capabilities over 68 ready models — with `AUTO`/`NPU`/`GPU` devices,
+async throughput, INT8, and your own detector trainable in the same package.
 
 [![CI](https://github.com/leeyunjai82/ovkit/actions/workflows/ci.yml/badge.svg)](https://github.com/leeyunjai82/ovkit/actions/workflows/ci.yml)
 [![Docs](https://img.shields.io/badge/docs-github.io-0a7d8c)](https://leeyunjai82.github.io/ovkit/)
@@ -38,6 +39,8 @@ for row in r.found:                    # [{'name': '사람', 'name_en': 'person'
 
 for r in Model("track", 0):            # webcam / video / "mic" -> a stream
     print(r, r.elapsed_ms, r.device)   # 2x person (#1, #4) 14.2 GPU
+    if not r.show("track"):            # a window; q or Esc ends the loop
+        break
 ```
 
 The same call also runs whole **capabilities** — several models chained into one
@@ -69,6 +72,8 @@ Python 3.10+. For development, install from source:
 git clone https://github.com/leeyunjai82/ovkit.git && cd ovkit
 python -m venv .venv && source .venv/bin/activate   # Windows: .venv\Scripts\activate
 pip install -e ".[dev]"
+
+python scripts/check.py --fix    # what CI runs: ruff, black, pytest
 ```
 
 ## Capabilities
@@ -93,6 +98,13 @@ camera index).
 | `attention` | `1 person looking at: laptop` | gaze + object detection (ray-cast into the boxes) |
 | `anonymize` | the picture with every face pixelated | face (and plate) detection + redaction |
 | `face_match` | `('yunjai', 0.81)` — who this is | embedding + cosine matching against your gallery |
+| `count` | `pencil 3 · cup 1` (optionally one kind only) | detection + per-kind tally |
+| `posture` | `neck 32° — sit up (6s)` | pose + neck angle **over time** |
+| `exercise` | `squat x 12 (down)` | pose + joint-angle hysteresis (squat, push-up) |
+| `attendance` | `present 24/26` + `roll.csv` | face detection + roster matching |
+| `teach` | your own classes, from example photos | embedding + cosine k-NN (5 modes) |
+| `depth` | `nearest: bottom-left · 34% of the frame is close` + a colour map | Depth Anything V2 Small |
+| `remove_background` | the subject, saved as a transparent PNG | U2-Net |
 
 ```python
 from ovkit import Model, list_pipelines
@@ -104,8 +116,78 @@ Model("face_analyze", attributes=("age_gender",))   # configure what runs
 ```
 
 Aliases: `ocr`, `anpr`, `blur`, `driver`, `describe`, `faces`, `people`, `vehicle`, `tracking`, `reid`.
-`ovkit capabilities` prints the list; **`ovkit gui` opens a window** where you can
-click through them against your webcam or a picture.
+
+## Teach your own AI (no GPU)
+
+A Teachable Machine in five lines — an embedding model turns examples into
+vectors, new inputs match the nearest ones. Five modes: `photo` (default),
+`face` (expressions), `hand` (needs `ovkit[hand]`), `upper`, `body`:
+
+```python
+ai = Model("teach")                       # or Model("가르치기")
+ai.learn("can", "photos/cans/")           # a folder per thing to recognise
+ai.learn("bottle", "photos/bottles/")
+print(ai.guess("new_photo.jpg"))          # ('can', 0.93)
+print(ai.score("test_photos/"))           # accuracy + what it confuses
+ai.save("recycling")                      # -> Documents/ovkit/recycling.json
+for r in ai.predict(0, stream=True): ...  # webcam over YOUR classes
+```
+
+Korean method names work too: `배우기`, `맞혀봐`, `점수`, `저장`. Collect
+examples with the webcam: `from ovkit.pipelines.teach import collect; collect("가위", 30)`.
+
+## Any model on the Hugging Face Hub
+
+The Open Model Zoo is an archive; the models people want are on the Hub.
+`optimum-intel` converts almost any of them to OpenVINO IR, so ovkit does not
+have to mirror a model to serve it:
+
+```bash
+pip install "ovkit[hf]"                      # the converter — needed once
+ovkit pull google/vit-base-patch16-224
+```
+
+```python
+Model("google/vit-base-patch16-224", "photo.jpg")   # tabby 0.94
+```
+
+Converting is the heavy half and happens once; what lands in the cache is a
+plain OpenVINO model, so **running it afterwards needs nothing but ovkit** — a
+classroom machine can be handed the cache and never see PyTorch. The pull also
+writes the model's own class names and the normalisation its image processor
+declares, so results read as answers and not as confident nonsense.
+
+Tasks: image classification, zero-shot (CLIP) classification, feature
+extraction. Detection and segmentation come from the registry instead
+(`detect`, `segment`) — optimum-intel does not export those.
+
+## Train your own detector
+
+RT-DETR, trainable, Apache-2.0 end to end — an Ultralytics-style workflow with
+no AGPL code or weights anywhere (`pip install "ovkit[train]"`):
+
+```python
+from ovkit import RTDETR
+
+model = RTDETR("rtdetr-r18")                 # COCO weights, downloaded on first use
+model.train(data="data.yaml", epochs=100)    # the YOLO-format labels you already have
+model.val(data="data.yaml").box.map50        # mAP50 / mAP50-95
+model.export(half=True)                      # -> IR + labels.txt
+```
+
+The three sizes are `rtdetr_r18` (20M, 46.4 COCO AP — what `detect` uses),
+`rtdetr_r34` (31M, 48.9) and `rtdetr_r50` (43M, 53.1), mirrored like every
+other model. Running them needs nothing but ovkit; `ovkit[train]` adds PyTorch
+for fine-tuning:
+
+```python
+Model("detect", "street.jpg")      # r18 — keeps up with a webcam
+Model("rtdetr_r50", "street.jpg")  # when you want the accuracy instead
+```
+
+A fine-tuned model's exported IR drops straight into `Model("path/to/best.xml")` — the
+`labels.txt` written next to it means your classes answer by name. CLI:
+`ovkit train --data data.yaml`, `ovkit val`, `ovkit export`.
 
 ## Supported tasks
 
@@ -124,7 +206,7 @@ Every single-model task below runs end-to-end through the same 3 lines — swap
 | LLM / STT (GenAI) | `llm`, `stt` | generated text | [llm.py](https://github.com/leeyunjai82/ovkit/blob/main/examples/llm.py) / [stt.py](https://github.com/leeyunjai82/ovkit/blob/main/examples/stt.py) |
 | NLP / audio / time series | `qa`, `translation`, `noise_suppression`, `time_series` | tensors via `model.infer()` | [denoise_audio.py](https://github.com/leeyunjai82/ovkit/blob/main/examples/denoise_audio.py) |
 
-The registry exposes **one well-tested model per capability (35 total)**; the
+The registry exposes **one well-tested model per capability (43 OpenVINO IR + 13 GenAI)**; the
 [HF mirror](https://huggingface.co/leeyunjai/ovkit-models) hosts the full
 Apache-2.0 OMZ set (other tiers, `int8`, `sparse` variants) — surfacing a
 variant is a one-line edit ([catalog](https://leeyunjai82.github.io/ovkit/models.html)).
@@ -216,7 +298,7 @@ ovkit devices                               # available OpenVINO devices
 ```python
 from ovkit.genai import pipeline
 
-llm = pipeline("llm")                        # tinyllama_chat from the mirror
+llm = pipeline("llm")                        # qwen25_1_5b_instruct from the mirror
 print(llm.generate("Explain OpenVINO in one sentence.", max_new_tokens=64))
 
 stt = pipeline("stt")                        # whisper_base
@@ -240,6 +322,40 @@ automatically and results render with overlays.
 
 </details>
 
+## The model mirror
+
+Everything ovkit downloads comes from one repository,
+[leeyunjai/ovkit-models](https://huggingface.co/leeyunjai/ovkit-models) — one
+copy for an offline site to take, one repository to keep alive. Models that
+originate elsewhere (the RT-DETR sizes, Depth Anything V2 Small, U2-Net) are
+copied in, and their home stays registered as the fallback.
+
+To refresh it after adding a model, run the **Sync the model mirror** workflow
+from the Actions tab (dry run by default; tick *upload* to copy). It needs an
+`HF_TOKEN` repository secret with write access, set once. The same thing runs
+locally:
+
+```bash
+export HF_TOKEN=hf_...
+python scripts/sync_mirror.py            # what is missing
+python scripts/sync_mirror.py --upload   # copy just that
+```
+
+It lists the target first and copies only what is absent, so re-running is safe.
+
+The traffic goes the other way too. A mirror only ever added to fills up with
+models that left the lineup and weights uploaded under an old path — and it is
+the thing a school clones whole:
+
+```bash
+python scripts/audit_mirror.py           # what no manifest references
+python scripts/audit_mirror.py --prune   # delete just that
+```
+
+A model's `README.md`, `LICENSE` and `labels.txt` stay as long as the model
+does, and go with it when it leaves. Deletions are ordinary Hub commits, so
+the history still has them.
+
 ## Adding a model
 
 Models are data, not code — one manifest entry

{ovkit-0.2.0 → ovkit-0.4.0}/examples/README.md

@@ -15,6 +15,7 @@ pip install -r examples/requirements.txt # fastapi / uvicorn (web demos only)
 | [`pose.py`](pose.py) | `pose` | keypoints/skeleton |
 | [`classify.py`](classify.py) | `classify` | top-5 classes |
 | [`ocr.py`](ocr.py) | `text_recognition` | decoded text |
+| [`speak.py`](speak.py) | `읽어주기` | text read aloud, saved as a WAV |
 | [`super_resolution.py`](super_resolution.py) | `super_resolution` | upscaled image |
 | [`background_matting.py`](background_matting.py) | `background_matting_mobilenetv2` | cut the subject out (needs frame + empty-scene photo) |
 

{ovkit-0.2.0 → ovkit-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "ovkit"
-version = "0.2.0"
+version = "0.4.0"
 description = "A simple Python inference API for OpenVINO: one import, one Model class, clean Results — with AUTO/NPU, async, and INT8."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -33,7 +33,18 @@ dependencies = [
 genai = ["openvino-genai", "optimum-intel"]
 anomaly = ["anomalib"]
 quant = ["nncf"]
-all = ["openvino-genai", "optimum-intel", "anomalib", "nncf"]
+train = ["rtdetr[train]>=0.6"]
+hand = ["mediapipe"]
+hf = ["optimum-intel[openvino]>=1.20", "transformers>=4.40"]
+all = [
+    "openvino-genai",
+    "optimum-intel[openvino]>=1.20",
+    "transformers>=4.40",
+    "anomalib",
+    "nncf",
+    "mediapipe",
+    "rtdetr[train]>=0.6",
+]
 dev = ["pytest>=7", "ruff", "black"]
 
 [project.urls]

{ovkit-0.2.0 → ovkit-0.4.0}/src/ovkit/__init__.py

@@ -24,6 +24,7 @@ whatever describes what was found. :func:`list_pipelines` shows them all.
 
 from __future__ import annotations
 
+from . import hub
 from .core.errors import (
     ConversionError,
     DownloadError,
@@ -40,10 +41,32 @@ from .core.registry import list_models
 from .core.results import Boxes, Keypoints, Masks, Probs, Results
 from .pipelines import Pipeline, list_pipelines
 
-__version__ = "0.2.0"
+__version__ = "0.4.0"
+
+
+def __getattr__(name: str):
+    """``ovkit.RTDETR`` re-exports the standalone ``rtdetr`` package.
+
+    Training lives in its own project (Apache-2.0 RT-DETR with an
+    Ultralytics-style API); ovkit re-exports it so a trained detector and the
+    capabilities sit behind one import. ``pip install "ovkit[train]"``.
+    """
+    if name == "RTDETR":
+        try:
+            from rtdetr import RTDETR
+        except ImportError as exc:
+            raise ImportError(
+                'Training needs the rtdetr package:  pip install "ovkit[train]"\n'
+                'Its exported IR runs here either way: Model("your-model.xml").'
+            ) from exc
+        return RTDETR
+    raise AttributeError(f"module 'ovkit' has no attribute {name!r}")
+
 
 __all__ = [
     "Model",
+    "hub",
+    "RTDETR",
     "Pipeline",
     "list_pipelines",
     "Results",