Upload folder using huggingface_hub
Browse files- README.md +6 -8
- docker_task_1/Dockerfile +6 -4
- docker_task_2/Dockerfile +9 -4
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 |
-
-
|
| 33 |
-
- `260_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Abdomen** region.
|
| 34 |
-
- `262_gt_nnUNetResEncUNetLPlans.json`: Normalization configuration for the **Head & Neck** region.
|
| 35 |
-
- `
|
| 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 |
-
|
| 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 |
-
|
| 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/
|