pyproject.toml

[tool.poetry]
name = "yolact"
version = "4.2.0"
description = "Fork of YOLACT"
authors = ["Psycle Research <tech@psycle.io>"]
license = "MIT"
readme = "README.md"
repository = "https://github.com/PsycleResearch/yolact"

include = [{ path = "yolact/**/*.so", format = "wheel" }]

[tool.poetry.build]
script = "build.py"
generate-setup-file = true

[[tool.poetry.source]]
name = "PyPI"
priority = "primary"

[[tool.poetry.source]]
name = "torchcpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"

[tool.poetry.dependencies]
python = "^3.9"
torch = { version = "^2.0.0", source = "torchcpu" }
torchvision = { version = ">=0.15,<1", source = "torchcpu" }
Cython = "*"
numpy = "^1.24.0"
pycocotools = "^2.0.2"
tqdm = "^4.48.2"
mashumaro = { version = "^3.0.1", extras = ["yaml"] }
opencv-python = "^4.4.0.42"

[tool.poetry.group.dev]

optional = true

[tool.poetry.group.dev.dependencies]
setuptools = ">=43.0"
scipy = "^1.5.2"
scikit-learn = ">=0.23.2,<2"
black = ">=22.8.0"
pylint = "^2.13.9"
h5py = "^3.1.0"

[build-system]
requires = ["poetry-core", "Cython", "numpy>=1.20.0,<2", "setuptools"]
build-backend = "poetry.core.masonry.api"