andreped commited on
Commit
224e8c1
1 Parent(s): f7fc178

fix for chainer inference when cupy is not installed and cpu is not specific

Browse files
Files changed (1) hide show
  1. livermask/utils/process.py +8 -0
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