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
|
- import sys
- from itertools import chain
- from deadtrees.constants import HOME_HTML, MODEL_CHECKPOINT_PATH, PACKAGE_DIR
- from deadtrees.version import __version__
- from setuptools import find_packages, setup
- if not MODEL_CHECKPOINT_PATH.exists():
- # develop will be in argv if we do e.g. `pip install -e .`
- if "develop" not in sys.argv:
- # logger.error("can't build a non-development package with no model")
- # raise FileNotFoundError(MODEL_CHECKPOINT_PATH)
- pass
- # extra package dependencies
- EXTRAS = {
- "train": ["wandb", "seaborn"],
- "preprocess": [
- "gdal",
- "pygeos",
- "bottleneck",
- "dask",
- "rioxarray>=0.4",
- "xarray",
- ],
- }
- EXTRAS["all"] = [i for i in chain.from_iterable(EXTRAS.values())]
- setup(
- name="deadtrees",
- version=__version__,
- packages=find_packages(),
- install_requires=[
- "albumentations",
- "dvc[s3]",
- "python-dotenv",
- "hydra-core>=1.1.0",
- "hydra-colorlog>=1.1.0",
- "pydantic",
- "torch>=1.10.0",
- "torchvision>=0.12.0",
- "pytorch-lightning>=1.5",
- "rich",
- "tqdm",
- "webdataset==0.1.62",
- "segmentation_models_pytorch>=0.2.1",
- "efficientnet-pytorch>=0.7.1",
- ],
- # install in editable mode: pip install -e ".[train,preprocess]" or
- # pip install -e ".[all]"
- extras_require=EXTRAS,
- entry_points={
- "demo": [
- "deadtrees=deadtrees.__main__:main",
- ],
- },
- package_data={
- "deadtrees": [
- # str(MODEL_CHECKPOINT_PATH.relative_to(PACKAGE_DIR) / "*.torch"),
- # str(HOME_HTML.relative_to(PACKAGE_DIR)),
- ]
- },
- )
|