Add pytorch-ignite for eval metrics computation

KolinGuo · Mar 8, 2021 · 333ac4f · 333ac4f
1 parent fa299df
commit 333ac4f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -161,7 +161,6 @@ ENV LC_ALL en_US.UTF-8
 # SECTION 6: Additional installations #
 #######################################
 # COCOAPI
-RUN python3 -m pip install cython
 RUN git clone https://github.com/cocodataset/cocoapi.git /usr/share/cocoapi && \
     cd /usr/share/cocoapi/PythonAPI && \
     sed -i 's/python/python3/g' Makefile && \

diff --git a/install/requirements.txt b/install/requirements.txt
@@ -9,6 +9,7 @@ pytest
 argparse
 pydot   # For tf.keras.utils.plot_model
 six     # For tf-nightly and tb-nightly
+cython
 
 # Image Processing
 Pillow
@@ -18,6 +19,7 @@ scikit-learn
 # ML Framework
 torch==1.8.0
 torchvision==0.9.0
+pytorch-ignite==0.4.4
 #tensorflow==2.4.0
 tensorboard
 tensorboard_plugin_profile