File size: 571 Bytes
eff52f6
3160d9f
4f9da36
3160d9f
4f9da36
 
 
 
 
3160d9f
4f9da36
 
 
3160d9f
4f9da36
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
FROM python:3.10.2-slim

RUN mkdir -p /repos/pyment

COPY scripts /repos/pyment/scripts
COPY pyment /repos/pyment/pyment
COPY pyproject.toml /repos/pyment/
COPY README.md /repos/pyment/
COPY LICENSE.md /repos/pyment/

RUN pip install --upgrade pip poetry-core build && \
    cd /repos/pyment && \
    pip install --no-cache-dir .

RUN mkdir -p /.pyment/weights && \
    chmod -R 1777 /.pyment
COPY checkpoints/pyment /.pyment/weights

CMD ["python", "/repos/pyment/scripts/predict_from_fastsurfer_folder.py", \
     "/fastsurfer", \
     "-d", "/output/predictions.csv"]