fix for chainer inference when cupy is not installed and cpu is not specific
Browse files
livermask/utils/process.py
CHANGED
@@ -131,6 +131,14 @@ def liver_segmenter(params):
|
|
131 |
|
132 |
def vessel_segmenter(curr, output, cpu, verbose, multiple_flag, liver_mask, name_vessel):
|
133 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
134 |
# load model
|
135 |
unet, xp = load_vessel_model(name_vessel, cpu)
|
136 |
|
|
|
131 |
|
132 |
def vessel_segmenter(curr, output, cpu, verbose, multiple_flag, liver_mask, name_vessel):
|
133 |
|
134 |
+
# check if cupy is available, if not, set cpu=True
|
135 |
+
try:
|
136 |
+
import cupy
|
137 |
+
except ModuleNotFoundError as e:
|
138 |
+
log.info(e)
|
139 |
+
log.info("cupy is not available. Setting cpu=True")
|
140 |
+
cpu = True
|
141 |
+
|
142 |
# load model
|
143 |
unet, xp = load_vessel_model(name_vessel, cpu)
|
144 |
|