FelixzeroSun commited on
Commit
25cbfe7
·
verified ·
1 Parent(s): 85a5b9d

Upload folder using huggingface_hub

Browse files
README.md CHANGED
@@ -29,15 +29,12 @@ Our methods emphasize **robust image synthesis**, **reproducible pipelines**, an
29
  Contains all necessary files to build and run the Docker image for **Task 2** (CBCT → sCT).
30
  - `process.py`: Script that performs inference, converting CBCT images into synthetic CT (sCT).
31
 
32
- - **Normalization Config Files**
33
- - `260_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Abdomen** region.
34
- - `262_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Head & Neck** region.
35
- - `266_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Thorax** region.
36
  These files are essential for inverse normalization, ensuring that the synthesized CT intensities are mapped back to their correct clinical ranges.
37
 
38
- - **`nnunet_results`**
39
- You also need to place your **trained models** under the `nnunet_results/` directory so that inference can correctly locate and load them.
40
-
41
  - **`Dockerfile`**
42
  Defines all steps and dependencies needed to build the Docker image. It ensures reproducibility and consistency across environments.
43
 
@@ -58,7 +55,8 @@ Our methods emphasize **robust image synthesis**, **reproducible pipelines**, an
58
 
59
 
60
  ## 🚀 Getting Started
61
-
 
62
  ### 1. Build the Docker Image
63
  ```bash
64
  cd docker_task_1
 
29
  Contains all necessary files to build and run the Docker image for **Task 2** (CBCT → sCT).
30
  - `process.py`: Script that performs inference, converting CBCT images into synthetic CT (sCT).
31
 
32
+ - **`Normalization Config Files`**
33
+ - `260_gt_nnUNetResEncUNetLPlans.json/540_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Abdomen** region.
34
+ - `262_gt_nnUNetResEncUNetLPlans.json/542_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Head & Neck** region.
35
+ - `264_gt_nnUNetResEncUNetLPlans.json/544_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Thorax** region.
36
  These files are essential for inverse normalization, ensuring that the synthesized CT intensities are mapped back to their correct clinical ranges.
37
 
 
 
 
38
  - **`Dockerfile`**
39
  Defines all steps and dependencies needed to build the Docker image. It ensures reproducibility and consistency across environments.
40
 
 
55
 
56
 
57
  ## 🚀 Getting Started
58
+ - **`nnunet_results`**
59
+ Before starting inference, you also need to create a folder called `nnunet_results/` in docker_task_1/docker_task_2 and place your **trained models** under the `nnunet_results/` directory so that inference can correctly locate and load them.
60
  ### 1. Build the Docker Image
61
  ```bash
62
  cd docker_task_1
docker_task_1/Dockerfile CHANGED
@@ -27,13 +27,15 @@ COPY --chown=algorithm:algorithm requirements.txt /opt/algorithm/
27
  RUN python -m pip install --user -r requirements.txt
28
  COPY --chown=algorithm:algorithm nnsyn /opt/algorithm/nnsyn
29
  RUN cd /opt/algorithm/nnsyn && python -m pip install --user -e .
30
-
 
 
31
  COPY --chown=algorithm:algorithm .env /opt/algorithm/
32
  COPY --chown=algorithm:algorithm process.py /opt/algorithm/
33
  COPY --chown=algorithm:algorithm base_algorithm.py /opt/algorithm/
34
- COPY --chown=algorithm:algorithm nnunet_preprocessed/ /opt/algorithm/nnunet_preprocessed
35
- COPY --chown=algorithm:algorithm nnunet_raw/ /opt/algorithm/nnunet_raw
36
- # COPY --chown=algorithm:algorithm nnunet_results/ /opt/algorithm/nnunet_results
37
  COPY --chown=algorithm:algorithm 260_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
38
  COPY --chown=algorithm:algorithm 262_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
39
  COPY --chown=algorithm:algorithm 264_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
 
27
  RUN python -m pip install --user -r requirements.txt
28
  COPY --chown=algorithm:algorithm nnsyn /opt/algorithm/nnsyn
29
  RUN cd /opt/algorithm/nnsyn && python -m pip install --user -e .
30
+ RUN mkdir -p /opt/algorithm/nnunet_raw \
31
+ /opt/algorithm/nnunet_preprocessed \
32
+ && chown -R algorithm:algorithm /opt/algorithm/nnunet_raw /opt/algorithm/nnunet_preprocessed
33
  COPY --chown=algorithm:algorithm .env /opt/algorithm/
34
  COPY --chown=algorithm:algorithm process.py /opt/algorithm/
35
  COPY --chown=algorithm:algorithm base_algorithm.py /opt/algorithm/
36
+ # COPY --chown=algorithm:algorithm nnunet_preprocessed/ /opt/algorithm/nnunet_preprocessed
37
+ # COPY --chown=algorithm:algorithm nnunet_raw/ /opt/algorithm/nnunet_raw
38
+ COPY --chown=algorithm:algorithm nnunet_results/ /opt/algorithm/nnunet_results
39
  COPY --chown=algorithm:algorithm 260_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
40
  COPY --chown=algorithm:algorithm 262_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
41
  COPY --chown=algorithm:algorithm 264_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
docker_task_2/Dockerfile CHANGED
@@ -21,7 +21,12 @@ ENV PATH="/home/algorithm/.local/bin:${PATH}"
21
  # ENV PYTHONPATH="/opt/algorithm/nnUNetv2:$PYTHONPATH"
22
 
23
  RUN python -m pip install --user -U pip
24
-
 
 
 
 
 
25
 
26
  COPY --chown=algorithm:algorithm requirements.txt /opt/algorithm/
27
  RUN python -m pip install --user -r requirements.txt
@@ -31,9 +36,9 @@ RUN cd /opt/algorithm/nnsyn && python -m pip install --user -e .
31
  COPY --chown=algorithm:algorithm .env /opt/algorithm/
32
  COPY --chown=algorithm:algorithm process.py /opt/algorithm/
33
  COPY --chown=algorithm:algorithm base_algorithm.py /opt/algorithm/
34
- COPY --chown=algorithm:algorithm nnunet_preprocessed/ /opt/algorithm/nnunet_preprocessed
35
- COPY --chown=algorithm:algorithm nnunet_raw/ /opt/algorithm/nnunet_raw
36
- # COPY --chown=algorithm:algorithm nnunet_results/ /opt/algorithm/nnunet_results
37
  COPY --chown=algorithm:algorithm 540_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
38
  COPY --chown=algorithm:algorithm 542_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
39
  COPY --chown=algorithm:algorithm 544_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
 
21
  # ENV PYTHONPATH="/opt/algorithm/nnUNetv2:$PYTHONPATH"
22
 
23
  RUN python -m pip install --user -U pip
24
+ RUN mkdir -p /opt/algorithm/nnunet_raw \
25
+ /opt/algorithm/nnunet_preprocessed \
26
+ && chown -R algorithm:algorithm /opt/algorithm/nnunet_raw /opt/algorithm/nnunet_preprocessed
27
+ COPY --chown=algorithm:algorithm .env /opt/algorithm/
28
+ COPY --chown=algorithm:algorithm process.py /opt/algorithm/
29
+ COPY --chown=algorithm:algorithm base_algorithm.py /opt/algorithm/
30
 
31
  COPY --chown=algorithm:algorithm requirements.txt /opt/algorithm/
32
  RUN python -m pip install --user -r requirements.txt
 
36
  COPY --chown=algorithm:algorithm .env /opt/algorithm/
37
  COPY --chown=algorithm:algorithm process.py /opt/algorithm/
38
  COPY --chown=algorithm:algorithm base_algorithm.py /opt/algorithm/
39
+ # COPY --chown=algorithm:algorithm nnunet_preprocessed/ /opt/algorithm/nnunet_preprocessed
40
+ # COPY --chown=algorithm:algorithm nnunet_raw/ /opt/algorithm/nnunet_raw
41
+ COPY --chown=algorithm:algorithm nnunet_results/ /opt/algorithm/nnunet_results
42
  COPY --chown=algorithm:algorithm 540_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
43
  COPY --chown=algorithm:algorithm 542_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/
44
  COPY --chown=algorithm:algorithm 544_gt_nnUNetResEncUNetLPlans.json /opt/algorithm/