Spaces:
Sleeping
Sleeping
File size: 2,379 Bytes
a3290d1 |
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 |
#!/usr/bin/env python
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
import os
from os import path
from setuptools import find_packages, setup
def get_version():
init_py_path = path.join(
path.abspath(path.dirname(__file__)), "comp2comp", "__init__.py"
)
init_py = open(init_py_path, "r").readlines()
version_line = [line.strip() for line in init_py if line.startswith("__version__")][
0
]
version = version_line.split("=")[-1].strip().strip("'\"")
# The following is used to build release packages.
# Users should never use it.
suffix = os.getenv("ABCTSEG_VERSION_SUFFIX", "")
version = version + suffix
if os.getenv("BUILD_NIGHTLY", "0") == "1":
from datetime import datetime
date_str = datetime.today().strftime("%y%m%d")
version = version + ".dev" + date_str
new_init_py = [line for line in init_py if not line.startswith("__version__")]
new_init_py.append('__version__ = "{}"\n'.format(version))
with open(init_py_path, "w") as f:
f.write("".join(new_init_py))
return version
setup(
name="comp2comp",
version=get_version(),
author="StanfordMIMI",
url="https://github.com/StanfordMIMI/Comp2Comp",
description="Computed tomography to body composition.",
packages=find_packages(exclude=("configs", "tests")),
python_requires=">=3.9",
install_requires=[
"pydicom",
"moviepy",
"numpy==1.23.5",
"h5py",
"tabulate",
"tqdm",
"silx",
"yacs",
"pandas",
"dosma",
"opencv-python",
"huggingface_hub",
"pycocotools",
"wget",
"tensorflow==2.12.0",
"totalsegmentator @ git+https://github.com/StanfordMIMI/TotalSegmentator.git",
"totalsegmentatorv2 @ git+https://github.com/StanfordMIMI/TotalSegmentatorV2.git",
],
extras_require={
"all": ["shapely", "psutil"],
"dev": [
# Formatting
"flake8",
"isort",
"black==22.8.0",
"flake8-bugbear",
"flake8-comprehensions",
# Docs
"mock",
"sphinx",
"sphinx-rtd-theme",
"recommonmark",
"myst-parser",
],
"contrast_phase": ["xgboost"],
},
)
|