diff --git a/backend/webserver/util/dextr.py b/backend/webserver/util/dextr.py index 234bc8be..5f1f03e4 100644 --- a/backend/webserver/util/dextr.py +++ b/backend/webserver/util/dextr.py @@ -1,6 +1,6 @@ from dextr import DEXTR -from ..config import Config +from config import Config model = DEXTR(nb_classes=1, resnet_layers=101, input_shape=(512, 512), weights_path=Config.DEXTR_FILE, - num_input_channels=4, classifier='psp', sigmoid=True) \ No newline at end of file + num_input_channels=4, classifier='psp', sigmoid=True)