Attempt to copy model inside /code container
Browse files- Dockerfile +1 -1
- livermask/utils/run.py +1 -0
Dockerfile
CHANGED
@@ -34,7 +34,7 @@ RUN apt install wget -y
|
|
34 |
|
35 |
# Download pretrained parenchyma model
|
36 |
RUN wget "https://github.com/andreped/livermask/releases/download/trained-models-v1/model.h5"
|
37 |
-
|
38 |
|
39 |
# Set up a new user named "user" with user ID 1000
|
40 |
RUN useradd -m -u 1000 user
|
|
|
34 |
|
35 |
# Download pretrained parenchyma model
|
36 |
RUN wget "https://github.com/andreped/livermask/releases/download/trained-models-v1/model.h5"
|
37 |
+
COPY --chown=user ./model.h5 /code/model.h5
|
38 |
|
39 |
# Set up a new user named "user" with user ID 1000
|
40 |
RUN useradd -m -u 1000 user
|
livermask/utils/run.py
CHANGED
@@ -70,6 +70,7 @@ def run_analysis(path, output, cpu, verbose, vessels, extension, name=None, name
|
|
70 |
if multiple_flag:
|
71 |
os.makedirs(output + "/", exist_ok=True)
|
72 |
|
|
|
73 |
for curr in tqdm(paths, "CT:"):
|
74 |
# check if current file is a nifti file, if not, skip
|
75 |
if curr.endswith(".nii") or curr.endswith(".nii.gz"):
|
|
|
70 |
if multiple_flag:
|
71 |
os.makedirs(output + "/", exist_ok=True)
|
72 |
|
73 |
+
log.info("Starting inference...")
|
74 |
for curr in tqdm(paths, "CT:"):
|
75 |
# check if current file is a nifti file, if not, skip
|
76 |
if curr.endswith(".nii") or curr.endswith(".nii.gz"):
|