init

.dockerignore

@@ -0,0 +1,6 @@
+.git
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.DS_Store

Dockerfile

@@ -0,0 +1,72 @@
+FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu22.04
+
+ARG DEBIAN_FRONTEND=noninteractive
+ARG PYTHON_VERSION=3.10
+ARG EGOFORCE_REPO_URL=https://github.com/Chris10M/EgoForce.git
+ARG EGOFORCE_REF=main
+
+ENV HOME=/home/user \
+    APP_HOME=/home/user/app \
+    EGOFORCE_ROOT=/home/user/app/EgoForce \
+    PATH=/home/user/.local/bin:/home/user/.venv/bin:$PATH \
+    PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=1 \
+    PIP_DISABLE_PIP_VERSION_CHECK=1 \
+    CUDA_HOME=/usr/local/cuda \
+    FORCE_CUDA=1 \
+    TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9" \
+    NVIDIA_VISIBLE_DEVICES=all \
+    NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics \
+    PYOPENGL_PLATFORM=egl \
+    MPLBACKEND=Agg
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.10 \
+    python3.10-dev \
+    python3-pip \
+    python3.10-venv \
+    git \
+    curl \
+    ca-certificates \
+    ffmpeg \
+    build-essential \
+    cmake \
+    ninja-build \
+    pkg-config \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender1 \
+    libgl1 \
+    libglvnd0 \
+    libegl1 \
+    libglu1-mesa \
+    libgl1-mesa-dev \
+    libegl1-mesa-dev \
+    libgles2-mesa-dev \
+    libgl1-mesa-dri \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN ln -sf /usr/bin/python3.10 /usr/local/bin/python && \
+    ln -sf /usr/bin/pip3 /usr/local/bin/pip
+
+RUN useradd -m -u 1000 user
+USER user
+
+WORKDIR $APP_HOME
+
+RUN python -m venv /home/user/.venv
+
+COPY --chown=user scripts/ $APP_HOME/scripts/
+
+RUN chmod +x $APP_HOME/scripts/install_egoforce.sh $APP_HOME/scripts/start.sh
+
+RUN git clone --depth 1 --branch "${EGOFORCE_REF}" "${EGOFORCE_REPO_URL}" "$EGOFORCE_ROOT"
+
+RUN bash $APP_HOME/scripts/install_egoforce.sh
+
+WORKDIR $EGOFORCE_ROOT
+
+EXPOSE 7860
+
+CMD ["bash", "/home/user/app/scripts/start.sh"]

README.md

@@ -3,12 +3,20 @@ title: EgoForce
 emoji: 🐠
 colorFrom: gray
 colorTo: purple
-sdk: gradio
-sdk_version: 6.9.0
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: false
 license: cc-by-nc-4.0
-short_description: 'Forearm-Guided Camera-Space 3D Hand Pose '
+short_description: Forearm-Guided Camera-Space 3D Hand Pose
+suggested_hardware: a10g-small
+startup_duration_timeout: 1h
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+This Space builds a Docker image that:
+
+- Clones `EgoForce` Repo.
+- Installs the CUDA-enabled Python and system dependencies required by the demo
+- Downloads the `_DATA` model assets from `chris10/EgoForce` on the Hugging Face Hub
+- Launches `demo/run_app.py` on port `7860`
+
+The runtime is Docker-based because the EgoForce demo depends on CUDA, Torch-TensorRT, MMCV, ffmpeg, and headless rendering support.

scripts/download_egoforce_data.py

@@ -0,0 +1,41 @@
+from __future__ import annotations
+
+import os
+import shutil
+from pathlib import Path
+
+from huggingface_hub import snapshot_download
+
+
+def main() -> None:
+    repo_root = Path(os.environ["EGOFORCE_ROOT"]).resolve()
+    target_data_dir = repo_root / "_DATA"
+    cache_root = repo_root / ".hf-download"
+
+    if target_data_dir.exists():
+        print(f"Using existing EgoForce assets at {target_data_dir}")
+        return
+
+    if cache_root.exists():
+        shutil.rmtree(cache_root)
+
+    snapshot_path = Path(
+        snapshot_download(
+            repo_id=os.environ.get("EGOFORCE_ASSETS_REPO_ID", "chris10/EgoForce"),
+            repo_type="model",
+            allow_patterns=["_DATA/**"],
+            local_dir=cache_root,
+        )
+    )
+    source_data_dir = snapshot_path / "_DATA"
+
+    if not source_data_dir.exists():
+        raise SystemExit(f"Downloaded snapshot did not contain {source_data_dir}")
+
+    shutil.copytree(source_data_dir, target_data_dir)
+    shutil.rmtree(cache_root, ignore_errors=True)
+    print(f"Downloaded EgoForce assets to {target_data_dir}")
+
+
+if __name__ == "__main__":
+    main()

scripts/install_egoforce.sh

@@ -0,0 +1,44 @@
+#!/usr/bin/env bash
+set -euxo pipefail
+
+: "${EGOFORCE_ROOT:?EGOFORCE_ROOT must be set}"
+
+cd "$EGOFORCE_ROOT"
+
+python -m pip install --upgrade pip setuptools==81.0.0 wheel cython pybind11
+python -m pip install torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cu126
+python -m pip install torch_tensorrt==2.8.0+cu126 --find-links https://download.pytorch.org/whl/torch-tensorrt
+python -m pip install -r scripts/requirements.txt
+python -m pip install mmcv==2.1.0 --no-build-isolation
+
+python -m pip install git+https://github.com/javrtg/AnyCalib.git --no-build-isolation
+python -m pip install git+https://github.com/mattloper/chumpy.git --no-build-isolation
+python -m pip install git+https://github.com/facebookresearch/pytorch3d.git --no-build-isolation
+
+python -m pip install ./thirdparty/datapipes
+python -m pip install ./thirdparty/mmdetection --no-build-isolation
+
+python -m pip install projectaria_client_sdk==1.1.0 --no-cache-dir
+python -m pip install huggingface_hub hf_xet
+
+python /home/user/app/scripts/download_egoforce_data.py
+
+python - <<'PY'
+from pathlib import Path
+
+repo_root = Path.cwd()
+required_paths = [
+    repo_root / "_DATA" / "model_weights.pth",
+    repo_root / "_DATA" / "epoch_460.pth",
+    repo_root / "_DATA" / "detector.torchscript",
+    repo_root / "_DATA" / "mano",
+]
+
+missing = [str(path) for path in required_paths if not path.exists()]
+if missing:
+    raise SystemExit(f"Missing required EgoForce assets after download: {missing}")
+
+print("Verified EgoForce assets in", repo_root / "_DATA")
+PY
+
+python -m pip install numpy==1.26.4

scripts/start.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+: "${EGOFORCE_ROOT:?EGOFORCE_ROOT must be set}"
+
+cd "$EGOFORCE_ROOT"
+
+if [[ ! -f demo/run_app.py ]]; then
+    echo "EgoForce demo entrypoint not found at $EGOFORCE_ROOT/demo/run_app.py" >&2
+    exit 1
+fi
+
+if [[ ! -d _DATA ]]; then
+    echo "EgoForce _DATA directory is missing at $EGOFORCE_ROOT/_DATA" >&2
+    exit 1
+fi
+
+exec python demo/run_app.py \
+    --server-name 0.0.0.0 \
+    --server-port "${PORT:-7860}"