armanid 2.0.0__py3-none-any.whl

armanid/__init__.py

@@ -0,0 +1,21 @@
+"""
+Armanid — Object Detection Library
+Creator  : Arman Guriyan
+Company  : Epic Rabbit
+Version  : 2.0.0
+"""
+__version__ = "2.0.0"
+__author__  = "Arman Guriyan"
+__company__ = "Epic Rabbit"
+__email__   = "arman@epicrabbit.io"
+__license__ = "MIT"
+
+from armanid.model   import Armanid
+from armanid.results import DetectionResult, BoundingBox
+from armanid.utils   import draw_detections, save_detections, run_webcam
+
+__all__ = [
+    "Armanid","DetectionResult","BoundingBox",
+    "draw_detections","save_detections","run_webcam",
+    "__version__","__author__","__company__",
+]

armanid/_engine/__init__.py

@@ -0,0 +1,48 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+__version__ = "8.4.31"
+
+import importlib
+import os
+from typing import TYPE_CHECKING
+
+# Set ENV variables (place before imports)
+if not os.environ.get("OMP_NUM_THREADS"):
+    os.environ["OMP_NUM_THREADS"] = "1"  # default for reduced CPU utilization during training
+
+from armanid._engine.utils import ASSETS, SETTINGS
+from armanid._engine.utils.checks import check_yolo as checks
+from armanid._engine.utils.downloads import download
+
+settings = SETTINGS
+
+MODELS = ("YOLO", "YOLOWorld", "YOLOE", "NAS", "SAM", "FastSAM", "RTDETR")
+
+__all__ = (
+    "__version__",
+    "ASSETS",
+    *MODELS,
+    "checks",
+    "download",
+    "settings",
+)
+
+if TYPE_CHECKING:
+    # Enable hints for type checkers
+    from ultralytics.models import YOLO, YOLOWorld, YOLOE, NAS, SAM, FastSAM, RTDETR  # noqa
+
+
+def __getattr__(name: str):
+    """Lazy-import model classes on first access."""
+    if name in MODELS:
+        return getattr(importlib.import_module("armanid._engine.models"), name)
+    raise AttributeError(f"module {__name__} has no attribute {name}")
+
+
+def __dir__():
+    """Extend dir() to include lazily available model names for IDE autocompletion."""
+    return sorted(set(globals()) | set(MODELS))
+
+
+if __name__ == "__main__":
+    print(__version__)