feat: Update setup for project

This commit is contained in:
kiennt
2025-08-14 09:27:02 +00:00
parent 34b17b0280
commit 33303aa62f
3 changed files with 4457 additions and 5 deletions

View File

@@ -1,6 +1,68 @@
[build-system]
requires = [
"setuptools>=62.3.0,<75.9",
"torch>=2.3.1",
]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "Grounded-SAM-2"
version = "1.0"
description = "Grounded SAM 2: Ground and Track Anything in Videos"
readme = "README.md"
requires-python = ">=3.10.0"
license = { text = "Apache 2.0" }
authors = [{ name = "Meta AI", email = "segment-anything@meta.com" }]
keywords = ["segmentation", "computer vision", "deep learning"]
dependencies = [
"torch>=2.3.1",
"torchvision>=0.18.1",
"numpy>=1.24.4",
"tqdm>=4.66.1",
"hydra-core>=1.3.2",
"iopath>=0.1.10",
"pillow>=9.4.0",
"opencv-python-headless>=4.11.0.86",
"supervision>=0.26.1",
"pycocotools>=2.0.10",
"transformers>=4.55.1",
"addict>=2.4.0",
"yapf>=0.43.0",
"timm>=1.0.19",
"pdf2image>=1.17.0",
]
[project.optional-dependencies]
notebooks = [
"matplotlib>=3.9.1",
"jupyter>=1.0.0",
"opencv-python>=4.7.0",
"eva-decord>=0.6.1",
]
interactive-demo = [
"Flask>=3.0.3",
"Flask-Cors>=5.0.0",
"av>=13.0.0",
"dataclasses-json>=0.6.7",
"eva-decord>=0.6.1",
"gunicorn>=23.0.0",
"imagesize>=1.4.1",
"pycocotools>=2.0.8",
"strawberry-graphql>=0.243.0",
]
dev = [
"black==24.2.0",
"usort==1.0.2",
"ufmt==2.0.0b2",
"fvcore>=0.1.5.post20221221",
"pandas>=2.2.2",
"scikit-image>=0.24.0",
"tensorboard>=2.17.0",
"pycocotools>=2.0.8",
"tensordict>=0.5.0",
"opencv-python>=4.7.0",
"submitit>=1.5.1",
]
[tool.setuptools]
# extensions = [{ name = "sam2._C", sources = ["sam2/csrc/connected_components.cu"] }]
packages = ["sam2", "grounding_dino"]