add ml master image for dicom work under leukoexpert

PHT-Medic · Apr 25, 2022 · 6a2a9c9 · 6a2a9c9
1 parent 281aa2d
commit 6a2a9c9
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/data/leuko-expert/ml/Dockerfile b/data/leuko-expert/ml/Dockerfile
@@ -0,0 +1,10 @@
+FROM python:3.9
+
+RUN apt-get update -yqq && \
+    apt-get dist-upgrade -yqq && \
+    apt-get install -yqq git
+
+COPY requirements.txt /tmp/requirements.txt
+
+RUN python3 -m pip install --upgrade pip
+RUN pip3 install -r /tmp/requirements.txt
diff --git a/data/leuko-expert/ml/README.md b/data/leuko-expert/ml/README.md
@@ -0,0 +1,5 @@
+# Leuko Expert docker image
+
+Docker image for the Leuko Expert use case of the.
+If Pytorch is needed and dicom images
+
diff --git a/data/leuko-expert/ml/requirements.txt b/data/leuko-expert/ml/requirements.txt
@@ -0,0 +1,10 @@
+minio
+numpy
+pandas
+scikit-learn
+skimage
+torch
+torchvision
+pydicom
+matplotlib
+tqdm