29 lines
786 B
Python
29 lines
786 B
Python
"""Run lightweight YOLO11s + RTMW-L door-risk inference."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
PACKAGE_DIR = Path(__file__).resolve().parent
|
|
PIPELINE = PACKAGE_DIR / "pipeline" / "rtmw_risk_pipeline.py"
|
|
|
|
|
|
def main() -> None:
|
|
environment = os.environ.copy()
|
|
python_paths = [str(PACKAGE_DIR / "pipeline")]
|
|
if environment.get("PYTHONPATH"):
|
|
python_paths.append(environment["PYTHONPATH"])
|
|
environment["PYTHONPATH"] = os.pathsep.join(python_paths)
|
|
environment["HF_HUB_OFFLINE"] = "1"
|
|
environment["PYTHONUTF8"] = "1"
|
|
command = [sys.executable, str(PIPELINE), *sys.argv[1:]]
|
|
raise SystemExit(subprocess.run(command, env=environment).returncode)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|