Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 35 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
[build-system]
requires = ["setuptools>=61", "wheel", "torch", "numpy"]
build-backend = "setuptools.build_meta"

[project]
name = "groundingdino"
version = "0.1.0"
description = "Open-set object detector (Grounding DINO)"
readme = "README.md"
license = { file = "LICENSE" }
authors = [
{ name = "International Digital Economy Academy" },
{ name = "Shilong Liu" }
]
requires-python = ">=3.9"

dependencies = [
"torch",
"torchvision",
"transformers",
"addict",
"yapf",
"timm",
"numpy<2",
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

numpy<2 because torch 2.1.2 (from dockerfile) is compiled with numpy 1:

  A module that was compiled using NumPy 1.x cannot be run in
  NumPy 2.2.6 as it may crash. To support both 1.x and 2.x
  versions of NumPy, modules must be compiled with NumPy 2.0.
  Some module may need to rebuild instead e.g. with 'pybind11>=2.12'.

  If you are a user of the module, the easiest solution will be to
  downgrade to 'numpy<2' or try to upgrade the affected module.
  We expect that some modules will need time to support NumPy 2.

"opencv-python",
"supervision>=0.22.0",
"pycocotools"
]

[project.urls]
Homepage = "https://github.com/IDEA-Research/GroundingDINO"
Source = "https://github.com/IDEA-Research/GroundingDINO"

[tool.setuptools.packages.find]
exclude = ["configs", "tests"]