-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
executable file
·37 lines (36 loc) · 1.14 KB
/
setup.py
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
from setuptools import setup, find_namespace_packages
setup(name='clNet',
packages=find_namespace_packages(include=["clnet", "clnet.*"]),
version='2.0.0',
description='Segment in the wild: life-long continual learning net (clNet)',
install_requires=[
"torch>=2.0.0",
"torchvision",
"tqdm",
"dicom2nifti",
"scikit-image>=0.14",
"medpy",
"opencv-python",
"albumentations",
"scipy>=1.7.0",
"pytorch-lightning",
"batchgenerators>=0.21",
"numpy",
"scikit-learn",
"SimpleITK",
"pandas",
"requests",
"nibabel",
"tifffile",
"matplotlib",
"einops",
"line_profiler",
],
entry_points={
'console_scripts': [
'clNet_plan_and_preprocess = clnet.experiment_planning.clnet_plan_and_preprocess:main',
'clNet_train = clnet.run.run_training:main',
'clNet_predict = clnet.inference.predict_simple:main',
],
}
)