andreped commited on
Commit
1ff1479
1 Parent(s): 82fe6a3

tried importing Config directly, instead of the script

Browse files
Files changed (1) hide show
  1. livermask/utils/process.py +2 -2
livermask/utils/process.py CHANGED
@@ -16,7 +16,7 @@ import logging as log
16
  import chainer
17
  import math
18
  from .unet3d import UNet3D
19
- import yaml_utils
20
  import yaml
21
  from tensorflow.keras import backend as K
22
  from numba import cuda
@@ -133,7 +133,7 @@ def vessel_segmenter(curr, output, cpu, verbose, multiple_flag, liver_mask, name
133
  unet, xp = load_vessel_model(name_vessel, cpu)
134
 
135
  # read config
136
- config = yaml_utils.Config(yaml.safe_load(open("./configs/base.yml")))
137
 
138
  # read data
139
  nib_volume = nib.load(curr)
 
16
  import chainer
17
  import math
18
  from .unet3d import UNet3D
19
+ from .yaml_utils import Config
20
  import yaml
21
  from tensorflow.keras import backend as K
22
  from numba import cuda
 
133
  unet, xp = load_vessel_model(name_vessel, cpu)
134
 
135
  # read config
136
+ config = Config(yaml.safe_load(open("./configs/base.yml")))
137
 
138
  # read data
139
  nib_volume = nib.load(curr)