diff --git a/models/bamf_ct_liver_tumor/config/default.yml b/models/bamf_ct_liver_tumor/config/default.yml index 4a6072ff..7520a706 100644 --- a/models/bamf_ct_liver_tumor/config/default.yml +++ b/models/bamf_ct_liver_tumor/config/default.yml @@ -25,7 +25,6 @@ modules: nnunet_config: 3d_fullres roi: SPLEEN,KIDNEY,GALLBLADDER,DUODENUM,PANCREAS,SMALL_INTESTINE,LUNG,LIVER,LIVER+TUMOR export_prob_maps: False - folds: 0 DsegConverter: model_name: bamf_ct_liver_tumor diff --git a/models/bamf_ct_liver_tumor/dockerfiles/Dockerfile b/models/bamf_ct_liver_tumor/dockerfiles/Dockerfile index 5fb6b907..c6a2973c 100644 --- a/models/bamf_ct_liver_tumor/dockerfiles/Dockerfile +++ b/models/bamf_ct_liver_tumor/dockerfiles/Dockerfile @@ -11,7 +11,7 @@ ENV SKLEARN_ALLOW_DEPRECATED_SKLEARN_PACKAGE_INSTALL=True RUN pip3 install --no-cache-dir nnunetv2==2.0 # Install updated segdb entries -RUN pip uninstall segdb +RUN pip uninstall segdb -y RUN pip install git+https://github.com/MHubAI/segdb.git # Clone the main branch of MHubAI/models