Introduce Focoos Open Source Models ✨

.devcontainer/cpu/devcontainer.json

@@ -9,7 +9,7 @@
     "remoteUser": "root",
     "workspaceFolder": "${localWorkspaceFolder}",
     "workspaceMount": "source=${localWorkspaceFolder},target=${localWorkspaceFolder},type=bind",
-    "postCreateCommand": "uv pip install -e .[cpu,dev]",
+    "postCreateCommand": "uv pip install -e .[onnx-cpu,dev]",
     "remoteEnv": {
         "UV_SYSTEM_PYTHON": "true"
     },

.devcontainer/cuda/devcontainer.json → .devcontainer/gpu/devcontainer.json

@@ -1,7 +1,7 @@
 {
     "name": "Focoos GPU (OnnxRuntime, CUDA)",
     "build": {
-        "target": "focoos-cuda",
+        "target": "focoos-gpu",
         "context": "../..",
         "dockerfile": "../../Dockerfile",
     },
@@ -12,7 +12,7 @@
     },
     "workspaceFolder": "${localWorkspaceFolder}",
     "workspaceMount": "source=${localWorkspaceFolder},target=${localWorkspaceFolder},type=bind",
-    "postCreateCommand": "uv pip install -e .[cuda,dev]",
+    "postCreateCommand": "uv pip install -e .[onnx,dev]",
     "remoteEnv": {
         "UV_SYSTEM_PYTHON": "true"
     },

.devcontainer/tensorrt/devcontainer.json

@@ -1,5 +1,5 @@
 {
-    "name": "Focoos GPU (OnnxRuntime, Torch, TensorRT)",
+    "name": "Focoos GPU (OnnxRuntime, TensorRT)",
     "build": {
         "context": "../..",
         "dockerfile": "../../Dockerfile",

.github/workflows/test.yml

@@ -5,8 +5,7 @@ on:
 env:
   UV_SYSTEM_PYTHON: 1
 jobs:
-  Run-test:
-
+  code-tests:
     permissions:
       id-token: write # This is required for requesting the JWT
       contents: read
@@ -30,7 +29,7 @@ jobs:
         with:
           python-version:  ${{ matrix.python-version }}
       - name: 📦 Install dev dependencies
-        run: uv pip install .[cpu,dev]
+        run: uv sync --extra dev --extra onnx-cpu
       - name: 🧪 Run test
         run: make test
       - name: 📊 Generate test coverage report
@@ -40,3 +39,41 @@ jobs:
           pytest-xml-coverage-path: ./tests/coverage.xml
           junitxml-path: ./tests/junit.xml
           report-only-changed-files: true
+          hide-report: true
+  model-tests:
+    runs-on: actions-runner-cuda12
+    steps:
+      - name: 📥 Checkout Repository
+        uses: actions/checkout@v4
+      - name: 🐍 Setup Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.12
+      - name: 🐍 Setup uv
+        uses: astral-sh/setup-uv@v4
+        with:
+          python-version: "3.12"
+          enable-cache: true
+          cache-dependency-glob: "uv.lock"
+      - name: 📦 Install dev dependencies
+        run: uv sync --extra dev --extra onnx
+      - name: 🧪 Run e2e tests for all models
+        run: |
+          echo "🔍 Discovering models in model registry..."
+          MODELS=$(find ./focoos/model_registry -maxdepth 1 -type f -name "*.json" | xargs -n1 basename | sed 's/\.json$//')
+
+          if [ -z "$MODELS" ]; then
+            echo "⚠️  No models found in registry"
+            exit 1
+          fi
+
+          echo "📋 Found models: $(echo "$MODELS" | tr '\n' ' ')"
+
+          for model in $MODELS; do
+            echo "🧪 Testing model: $model"
+            uv run ops/test_training.py --model "$model"
+          done
+
+          echo "✅ All model tests completed successfully"
+      - name: 🧹 Minimize uv cache
+        run: uv cache prune --ci

.gitignore

@@ -91,4 +91,10 @@ notebooks/.data
 /data
 tests/junit.xml
 notebooks/datasets
+notebooks/experiments
 site/
+/datasets/
+/examples/
+notebooks/test.ipynb
+gradio/output/
+tutorials/experiments

Dockerfile

@@ -1,36 +1,24 @@
 FROM python:3.12-slim-bullseye AS focoos-cpu
 COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 LABEL authors="focoos.ai"
-RUN apt-get update && \
+RUN apt-get update --fix-missing && \
     apt-get install -y --no-install-recommends \
     build-essential git ffmpeg libsm6 libxext6 gcc libmagic1 wget make cmake
 WORKDIR /app
 COPY focoos ./focoos
 COPY pyproject.toml ./pyproject.toml
-RUN uv pip install --system -e .[cpu]
+RUN uv pip install --system -e .[onnx-cpu]
 
 
-FROM ghcr.io/focoosai/deeplearning:base-cu12-cudnn9-py312-uv AS focoos-cuda
+FROM ghcr.io/focoosai/deeplearning:base-cu12-cudnn9-py312-uv AS focoos-gpu
 LABEL authors="focoos.ai"
 
 WORKDIR /app
 
 COPY focoos ./focoos
 COPY pyproject.toml ./pyproject.toml
-RUN uv pip install --system -e .[cuda]
+RUN uv pip install --system -e .[onnx]
 
 
-FROM focoos-cuda AS focoos-torch
-RUN uv pip install --system -e .[torch]
-
-FROM focoos-torch AS focoos-tensorrt
-RUN apt-get update && apt-get install -y \
-    wget lsb-release && \
-    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb && \
-    dpkg -i cuda-keyring_1.1-1_all.deb  && \
-    apt-get update && apt-get install -y \
-    tensorrt \
-    python3-libnvinfer-dev \
-    uff-converter-tf && \
-    apt-get clean && rm -rf /var/lib/apt/lists/*
+FROM focoos-gpu AS focoos-tensorrt
 RUN uv pip install --system -e .[tensorrt]

Makefile

@@ -10,18 +10,20 @@ venv:
 	@uv venv --python=python3.12
 
 install: .uv .pre-commit
-	@uv pip install -e ".[dev,docs]" --no-cache-dir
+	@uv sync --extra onnx --extra tensorrt --extra dev --extra docs
 	@pre-commit install
 
-install-gpu: .uv .pre-commit
-	@uv pip install -e ".[dev,cuda,tensorrt,torch,docs]" --no-cache-dir
+install-cpu: .uv .pre-commit
+	@uv sync --extra onnx-cpu --extra dev --extra docs
 	@pre-commit install
 
+
+
 docs:
-	@mkdocs build --clean
+	@uv run mkdocs build --clean
 
 serve-docs:
-	@mkdocs serve
+	@uv run mkdocs serve
 
 lint:
 	@ruff check ./focoos ./tests ./notebooks  --fix
@@ -31,7 +33,7 @@ run-pre-commit: .pre-commit
 	@pre-commit run --all-files
 
 test:
-	@pytest -s --cov=focoos --cov-report="xml:tests/coverage.xml" --cov-report=html --junitxml=./tests/junit.xml   && rm -f .coverage
+	@uv run pytest -s tests --cov=focoos --cov-report=term-missing:skip-covered --cov-report="xml:tests/coverage.xml" --cov-report=html --junitxml=./tests/junit.xml && rm -f .coverage
 
 tox:
 	tox