diff --git a/models/bamf_nnunet_ct_lung/config/default.yml b/models/bamf_nnunet_ct_lungnodules/config/default.yml similarity index 92% rename from models/bamf_nnunet_ct_lung/config/default.yml rename to models/bamf_nnunet_ct_lungnodules/config/default.yml index 19d149b3..1564bee2 100644 --- a/models/bamf_nnunet_ct_lung/config/default.yml +++ b/models/bamf_nnunet_ct_lungnodules/config/default.yml @@ -33,4 +33,4 @@ modules: DataOrganizer: targets: - - dicomseg-->[i:sid]/bamf_ct_lung_nodule.seg.dcm + - dicomseg-->[i:sid]/bamf_nnunet_ct_lungnodules.seg.dcm diff --git a/models/bamf_nnunet_ct_lung/dockerfiles/Dockerfile b/models/bamf_nnunet_ct_lungnodules/dockerfiles/Dockerfile similarity index 85% rename from models/bamf_nnunet_ct_lung/dockerfiles/Dockerfile rename to models/bamf_nnunet_ct_lungnodules/dockerfiles/Dockerfile index f8d234d0..f7ebf9eb 100644 --- a/models/bamf_nnunet_ct_lung/dockerfiles/Dockerfile +++ b/models/bamf_nnunet_ct_lungnodules/dockerfiles/Dockerfile @@ -12,7 +12,7 @@ RUN pip3 install --no-cache-dir nnunetv2==2.0 # Clone the main branch of MHubAI/models ARG MHUB_MODELS_REPO -RUN buildutils/import_mhub_model.sh bamf_ct_lung_nodule ${MHUB_MODELS_REPO} +RUN buildutils/import_mhub_model.sh bamf_nnunet_ct_lungnodules ${MHUB_MODELS_REPO} # Pull weights into the container ENV WEIGHTS_DIR=/root/.nnunet/nnUNet_models/nnUNet/ @@ -28,4 +28,4 @@ ENV WEIGHTS_FOLDER=$WEIGHTS_DIR # Default run script ENTRYPOINT ["mhub.run"] -CMD ["--config", "/app/models/bamf_ct_lung_nodule/config/default.yml"] +CMD ["--config", "/app/models/bamf_nnunet_ct_lungnodules/config/default.yml"] diff --git a/models/bamf_nnunet_ct_lung/meta.json b/models/bamf_nnunet_ct_lungnodules/meta.json similarity index 99% rename from models/bamf_nnunet_ct_lung/meta.json rename to models/bamf_nnunet_ct_lungnodules/meta.json index 0a7e255c..56e8f0e3 100644 --- a/models/bamf_nnunet_ct_lung/meta.json +++ b/models/bamf_nnunet_ct_lungnodules/meta.json @@ -1,6 +1,6 @@ { "id": "", - "name": "bamf_nnunet_ct_lung", + "name": "bamf_nnunet_ct_lungnodules", "title": "AIMI CT Lung and Nodules", "summary": { "description": "An nnU-Net based model to segment Lung and Nodules (3mm-30mm) from CT scans", diff --git a/models/bamf_nnunet_ct_lung/utils/BamfProcessorRunner.py b/models/bamf_nnunet_ct_lungnodules/utils/BamfProcessorRunner.py similarity index 100% rename from models/bamf_nnunet_ct_lung/utils/BamfProcessorRunner.py rename to models/bamf_nnunet_ct_lungnodules/utils/BamfProcessorRunner.py diff --git a/models/bamf_nnunet_ct_lung/utils/NNUnetRunnerV2.py b/models/bamf_nnunet_ct_lungnodules/utils/NNUnetRunnerV2.py similarity index 100% rename from models/bamf_nnunet_ct_lung/utils/NNUnetRunnerV2.py rename to models/bamf_nnunet_ct_lungnodules/utils/NNUnetRunnerV2.py diff --git a/models/bamf_nnunet_ct_lung/utils/__init__.py b/models/bamf_nnunet_ct_lungnodules/utils/__init__.py similarity index 100% rename from models/bamf_nnunet_ct_lung/utils/__init__.py rename to models/bamf_nnunet_ct_lungnodules/utils/__init__.py