pytorch-image-segmentation-lab/pyproject.toml at main · Doithoo/pytorch-image-segmentation-lab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
[build-system]
requires = ["setuptools>=83,<85"]
build-backend = "setuptools.build_meta"

[project]
name = "image-segmenter"
version = "0.1.0"
description = "Beginner-oriented reproducible PyTorch semantic segmentation"
readme = "README.md"
requires-python = ">=3.10,<3.13"
license = "MIT"
license-files = ["LICENSE"]
authors = [{ name = "Yashowhoo" }]
keywords = ["pytorch", "semantic-segmentation", "unet", "computer-vision", "education"]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Education",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "torch>=2.6,<3",
    "torchvision>=0.21,<1",
    "numpy>=1.24,<3",
    "pillow>=12.3,<13",
    "pyyaml>=6,<7",
]

[project.optional-dependencies]
dev = [
    "pytest>=9.0.3,<10",
    "ruff>=0.6,<1",
    "pre-commit>=3,<5",
    "build>=1.2,<2",
    "setuptools>=83,<85",
    "twine>=7.0.0,<8",
    "matplotlib>=3.6,<4",
    "mypy>=2.3.1,<3",
    "types-PyYAML>=6,<7",
    "tomli>=2,<3; python_version < '3.11'",
]
plot = ["matplotlib>=3.6,<4"]
model-zoo = ["segmentation-models-pytorch>=0.5,<1"]

[project.urls]
Homepage = "https://github.com/Doithoo/pytorch-image-segmentation-lab"
Repository = "https://github.com/Doithoo/pytorch-image-segmentation-lab"
Documentation = "https://github.com/Doithoo/pytorch-image-segmentation-lab#readme"
Source = "https://github.com/Doithoo/pytorch-image-segmentation-lab"
Issues = "https://github.com/Doithoo/pytorch-image-segmentation-lab/issues"

[project.scripts]
segment = "image_segmenter.cli:main"

[tool.setuptools.packages.find]
where = ["src"]

[tool.setuptools.package-data]
image_segmenter = ["py.typed"]

[tool.setuptools.data-files]
"share/image-segmenter/configs" = [
    "configs/deeplabv3_resnet50.yaml",
    "configs/fcn_resnet50.yaml",
    "configs/learning_minimal.yaml",
    "configs/lraspp_mobilenet.yaml",
    "configs/reference_unet.yaml",
    "configs/smp_unetplusplus.yaml",
    "configs/voc2012.yaml",
]

[tool.ruff]
line-length = 120
target-version = "py310"
src = ["src", "tests"]

[tool.ruff.lint]
select = ["E", "F", "W", "I", "UP", "B", "SIM"]
ignore = ["E501"]

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "-q -m 'not model_zoo'"
pythonpath = ["."]
markers = ["model_zoo: tests requiring the optional model-zoo dependency"]

[tool.mypy]
python_version = "3.12"
files = ["src", "scripts", "examples"]
check_untyped_defs = true
no_implicit_optional = true
warn_unused_ignores = true

[[tool.mypy.overrides]]
module = ["segmentation_models_pytorch", "segmentation_models_pytorch.*", "torchvision", "torchvision.*"]
ignore_missing_imports = true