add support for LITS and MSCOCO

youtubevos · Jul 20, 2018 · f649e02 · f649e02
1 parent a9ca725
commit f649e02
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -30,5 +30,16 @@ RUN apt-get update && apt-get install -y --fix-missing openjdk-7-jre r-base mono
 
 # Helpful python libraries
 RUN pip install \
-    nibabel==2.1.0 \
-    theano==0.9.0
+        nibabel==2.1.0 \
+        theano==0.9.0 \
+        Cython==0.27.3
+
+# MSCOCO dependencies
+RUN git clone https://github.com/cocodataset/cocoapi.git && \
+    cd cocoapi/PythonAPI && \
+    make && \
+    python setup.py build_ext install
+
+# LITS dependencies
+RUN pip install \
+        medpy==0.3.0
diff --git a/README.md b/README.md
@@ -16,8 +16,8 @@ This repo stores the legacy codalab worker, which has support for:
 
 ## Updating the image
 
-```docker build -t ckcollab/codalab-legacy:latest .```
+```docker build -t codalab/codalab-legacy:1.0.0 .```
 
 ## Pushing to dockerhub
 
-```docker push ckcollab/codalab-legacy```
+```docker push codalab/codalab-legacy```