-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
34 lines (33 loc) · 938 Bytes
/
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
from setuptools import setup, find_namespace_packages
setup(
name='RS2',
version='1.0',
description='deep learn based skull stripping tool for rodent fmri',
author='Yongkang Lin',
license=' GNU GENERAL PUBLIC LICENSE Version 3, 29 June 2007',
python_requires=">=3.9",
packages=find_namespace_packages(
include=['RS2', 'RS2.*']),
install_requires=[
"torch==2.0.0",
'monai>=1.1.0',
"nibabel",
"tqdm",
"einops",
"numpy",
"tifffile",
"batchgenerators",
"simpleitk",
"scipy",
"imageio",
"pandas",
"acvl_utils"
],
entry_points={
'console_scripts': [
'RS2_predict = RS2.inference.predict:predict_entry_point', # api available
],
},
keywords=['deep learning', 'image segmentation',
'medical image segmentation', 'mouse brain', 'rat brain']
)