Skip to content

Commit 834b4df

Browse files
authored
hotfix checkpoint download in docker
Merge pull request #326 from Deep-MI/hotfix/Dockerfiles_segonly_ckpt_download Added CerebNet, removed copying empty dir and corrected ckpt downloaddir
2 parents c951b24 + d073482 commit 834b4df

File tree

3 files changed

+9
-8
lines changed

3 files changed

+9
-8
lines changed

Docker/Dockerfile_FastSurferCNN

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,13 +63,13 @@ SHELL ["/bin/bash", "--login", "-c"]
6363
# Copy venv and FastSurferCNN
6464
COPY --from=build /venv /venv
6565
COPY ./FastSurferCNN /fastsurfer/FastSurferCNN/
66-
COPY ./checkpoints /fastsurfer/FastSurferCNN/checkpoints/
66+
COPY ./CerebNet /fastsurfer/CerebNet
6767
COPY ./Docker/entrypoint.sh /fastsurfer/Docker/entrypoint.sh
6868

6969
# Download all remote network checkpoints already
7070
ENV PYTHONPATH=/fastsurfer:$PYTHONPATH \
7171
MPLCONFIGDIR=/tmp
72-
RUN python3 /fastsurfer/FastSurferCNN/download_checkpoints.py --all; chmod +r -R /fastsurfer/FastSurferCNN/checkpoints
72+
RUN python3 /fastsurfer/FastSurferCNN/download_checkpoints.py --all
7373

7474
# Set FastSurfer workdir and entrypoint
7575
# the script entrypoint ensures that our conda env is active

Docker/Dockerfile_FastSurferCNN_AMD

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,14 @@ RUN pip3 install \
2020

2121

2222
# Copy FastSurferCNN
23-
COPY ./FastSurferCNN /FastSurferCNN/
23+
COPY ./FastSurferCNN /fastsurfer/FastSurferCNN/
24+
COPY ./CerebNet /fastsurfer/CerebNet
2425

2526
# Download all remote network checkpoints already
26-
RUN cd /FastSurferCNN/ ; python3 download_checkpoints.py --all --vinn
27+
RUN python3 /fastsurfer/FastSurferCNN/download_checkpoints.py --all
2728

2829
# Set FastSurferCNN workdir and entrypoint
2930
# the script entrypoint ensures that our conda env is active
30-
WORKDIR "/FastSurferCNN"
31-
ENTRYPOINT ["python3.7", "run_prediction.py"]
31+
WORKDIR "/fastsurfer"
32+
ENTRYPOINT ["python3.7", "./FastSurferCNN/run_prediction.py"]
3233
CMD ["--help"]

Docker/Dockerfile_FastSurferCNN_CPU

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,13 +63,13 @@ SHELL ["/bin/bash", "--login", "-c"]
6363
# Copy venv and FastSurferCNN
6464
COPY --from=build /venv /venv
6565
COPY ./FastSurferCNN /fastsurfer/FastSurferCNN/
66-
COPY ./checkpoints /fastsurfer/FastSurferCNN/checkpoints/
66+
COPY ./CerebNet /fastsurfer/CerebNet
6767
COPY ./Docker/entrypoint.sh /fastsurfer/Docker/entrypoint.sh
6868

6969
# Download all remote network checkpoints already
7070
ENV PYTHONPATH=/fastsurfer:$PYTHONPATH \
7171
MPLCONFIGDIR=/tmp
72-
RUN python3 /fastsurfer/FastSurferCNN/download_checkpoints.py --all; chmod +r -R /fastsurfer/FastSurferCNN/checkpoints
72+
RUN python3 /fastsurfer/FastSurferCNN/download_checkpoints.py --all
7373

7474
# Set FastSurfer workdir and entrypoint
7575
# the script entrypoint ensures that our conda env is active

0 commit comments

Comments
 (0)