tried importing Config directly, instead of the script
Browse files
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
|
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 =
|
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)
|