update name and added comment regarding GPU
Browse files- livermask/livermask.py +1 -6
- setup.py +1 -1
livermask/livermask.py
CHANGED
|
@@ -117,7 +117,7 @@ def func(path, output):
|
|
| 117 |
|
| 118 |
|
| 119 |
def main():
|
| 120 |
-
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
| 121 |
|
| 122 |
path = sys.argv[1]
|
| 123 |
output = sys.argv[2]
|
|
@@ -127,8 +127,3 @@ def main():
|
|
| 127 |
|
| 128 |
if __name__ == "__main__":
|
| 129 |
main()
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
-
|
|
|
|
| 117 |
|
| 118 |
|
| 119 |
def main():
|
| 120 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # disable GPU
|
| 121 |
|
| 122 |
path = sys.argv[1]
|
| 123 |
output = sys.argv[2]
|
|
|
|
| 127 |
|
| 128 |
if __name__ == "__main__":
|
| 129 |
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
setup.py
CHANGED
|
@@ -6,7 +6,7 @@ with open("README.md", "r") as f:
|
|
| 6 |
setuptools.setup(
|
| 7 |
name='livermask',
|
| 8 |
version='1.0.0',
|
| 9 |
-
author="
|
| 10 |
author_email="[email protected]",
|
| 11 |
license='MIT',
|
| 12 |
description="A package for automatic segmentation of liver from CT data",
|
|
|
|
| 6 |
setuptools.setup(
|
| 7 |
name='livermask',
|
| 8 |
version='1.0.0',
|
| 9 |
+
author="André Pedersen",
|
| 10 |
author_email="[email protected]",
|
| 11 |
license='MIT',
|
| 12 |
description="A package for automatic segmentation of liver from CT data",
|