Skip to content

Commit

Permalink
change folder name
Browse files Browse the repository at this point in the history
  • Loading branch information
jithenece committed Jul 30, 2024
1 parent 1d77d3a commit c9a1cfe
Show file tree
Hide file tree
Showing 6 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Original file line number Diff line number Diff line change
Expand Up @@ -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/
Expand All @@ -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"]
Original file line number Diff line number Diff line change
@@ -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",
Expand Down

0 comments on commit c9a1cfe

Please sign in to comment.