diff --git a/pyproject-cpu.toml b/pyproject-cpu.toml index 50348fa..8c73c04 100644 --- a/pyproject-cpu.toml +++ b/pyproject-cpu.toml @@ -1,7 +1,7 @@ [project] name = "winnow" -version = "0.2.13" -description = "Immich to Frigate training sets" +version = "0.4.10" +description = "Selects diverse, high-quality photos from Immich as training data for Frigate face recognition." license = "AGPL-3.0-or-later" requires-python = ">=3.13" authors = [{ name = "Holden Salomon", email = "holden@arch.fyi" }] @@ -23,10 +23,6 @@ dependencies = [ "python-dotenv>=1.2.1", "requests>=2.32.5", "rich>=14.2.0", - "torch>=2.12.0", - "torchvision>=0.27.0", - "transformers>=5.12.0", - "ultralytics>=8.4.66", ] [project.scripts] @@ -34,25 +30,13 @@ winnow = "winnow.cli:main" [project.urls] Repository = "https://github.com/sudolulo/winnow" +Changelog = "https://github.com/sudolulo/winnow/blob/main/CHANGELOG.md" [tool.uv] required-environments = [ "sys_platform == 'linux' and platform_machine == 'x86_64'", ] -[tool.uv.sources] -torch = [ - { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, -] -torchvision = [ - { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, -] - -[[tool.uv.index]] -name = "pytorch-cpu" -url = "https://download.pytorch.org/whl/cpu" -explicit = true - [dependency-groups] dev = [ "pytest>=8.0", @@ -81,9 +65,6 @@ numpy = "numpy" onnxruntime = "onnxruntime" requests = "requests" rich = "rich" -torch = "torch" -transformers = "transformers" -ultralytics = "ultralytics" [tool.pytest.ini_options] testpaths = ["tests"] diff --git a/pyproject-intel.toml b/pyproject-intel.toml index 3cedf81..e50234b 100644 --- a/pyproject-intel.toml +++ b/pyproject-intel.toml @@ -1,7 +1,7 @@ [project] name = "winnow" -version = "0.2.13" -description = "Immich to Frigate training sets" +version = "0.4.10" +description = "Selects diverse, high-quality photos from Immich as training data for Frigate face recognition." license = "AGPL-3.0-or-later" requires-python = ">=3.13" authors = [{ name = "Holden Salomon", email = "holden@arch.fyi" }] @@ -23,10 +23,6 @@ dependencies = [ "python-dotenv>=1.2.1", "requests>=2.32.5", "rich>=14.2.0", - "torch>=2.12.0", - "torchvision>=0.27.0", - "transformers>=5.12.0", - "ultralytics>=8.4.66", ] [project.scripts] @@ -34,6 +30,7 @@ winnow = "winnow.cli:main" [project.urls] Repository = "https://github.com/sudolulo/winnow" +Changelog = "https://github.com/sudolulo/winnow/blob/main/CHANGELOG.md" [tool.uv] conflicts = [ @@ -47,19 +44,6 @@ required-environments = [ "sys_platform == 'linux' and platform_machine == 'x86_64'", ] -[tool.uv.sources] -torch = [ - { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, -] -torchvision = [ - { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, -] - -[[tool.uv.index]] -name = "pytorch-cpu" -url = "https://download.pytorch.org/whl/cpu" -explicit = true - [dependency-groups] dev = [ "pytest>=8.0", @@ -88,9 +72,6 @@ numpy = "numpy" onnxruntime-openvino = "onnxruntime" requests = "requests" rich = "rich" -torch = "torch" -transformers = "transformers" -ultralytics = "ultralytics" [tool.deptry.per_rule_ignores] DEP002 = ["onnxruntime-openvino"] diff --git a/pyproject-rocm.toml b/pyproject-rocm.toml index 289edfd..7600361 100644 --- a/pyproject-rocm.toml +++ b/pyproject-rocm.toml @@ -1,7 +1,7 @@ [project] name = "winnow" -version = "0.2.13" -description = "Immich to Frigate training sets" +version = "0.4.10" +description = "Selects diverse, high-quality photos from Immich as training data for Frigate face recognition." license = "AGPL-3.0-or-later" requires-python = ">=3.13" authors = [{ name = "Holden Salomon", email = "holden@arch.fyi" }] @@ -23,10 +23,6 @@ dependencies = [ "python-dotenv>=1.2.1", "requests>=2.32.5", "rich>=14.2.0", - "torch>=2.5.0", - "torchvision>=0.20.0", - "transformers>=5.12.0", - "ultralytics>=8.4.66", ] [project.scripts] @@ -34,6 +30,7 @@ winnow = "winnow.cli:main" [project.urls] Repository = "https://github.com/sudolulo/winnow" +Changelog = "https://github.com/sudolulo/winnow/blob/main/CHANGELOG.md" [tool.uv] index-strategy = "unsafe-best-match" @@ -48,18 +45,6 @@ required-environments = [ "sys_platform == 'linux' and platform_machine == 'x86_64'", ] -[tool.uv.sources] -torch = [ - { index = "pytorch-rocm63", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, -] -torchvision = [ - { index = "pytorch-rocm63", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, -] - -[[tool.uv.index]] -name = "pytorch-rocm63" -url = "https://download.pytorch.org/whl/rocm6.3" - [dependency-groups] dev = [ "pytest>=8.0", @@ -88,9 +73,6 @@ numpy = "numpy" onnxruntime-rocm = "onnxruntime" requests = "requests" rich = "rich" -torch = "torch" -transformers = "transformers" -ultralytics = "ultralytics" [tool.deptry.per_rule_ignores] DEP002 = ["onnxruntime-rocm"] diff --git a/winnow/immich_api.py b/winnow/immich_api.py index 59cac53..d3332be 100644 --- a/winnow/immich_api.py +++ b/winnow/immich_api.py @@ -108,7 +108,7 @@ def fetch_all_assets(person: dict) -> list[dict]: def fetch_face_data(asset_id: str, person_id: str | None = None) -> FaceData | None: - """Fetch pre-computed face data (embedding, bbox, confidence) from Immich. + """Fetch pre-computed face data (bbox, confidence) from Immich. Queries GET /api/faces?id={asset_id} to retrieve face detection results that Immich already computed using InsightFace Buffalo_L.