Add requirements.txt to install dependencies in Docker container

Dockerfile

@@ -29,24 +29,14 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
     rm get-pip.py
 
 # Install Python dependencies
-RUN pip install -U \
-      numpy \
-      matplotlib \
-      seaborn \
-      pandas \
-      tqdm \
-      pillow \
-      setuptools \
-      sklearn \
-      scipy \
-      visdom
+COPY requirements.txt /root/MagnetLoss/requirements.txt
+WORKDIR /root/MagnetLoss
+RUN pip install --upgrade -r requirements.txt
 
 # Install PyTorch
 RUN pip install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp27-cp27mu-linux_x86_64.whl
 RUN pip install torchvision
 
 COPY ./ /root/MagnetLoss
 
-WORKDIR /root/MagnetLoss
-
 CMD /bin/bash

README.md

@@ -7,9 +7,8 @@ PyTorch implementation of the Magnet Loss for Deep Metric Learning, based on the
 ## Table of Contents
 * [Installation](#installation)
 * [Anaconda](#anaconda)
-* [Docker](#docker)
+* [Docker GPU Training](#docker-gpu-training)
 * [Results](#results)
-* [Other Implementations](#other-implementations)
 * [Citing MagnetLoss-PyTorch](#citing-magnetloss-pytorch)
 
 ## Installation
@@ -89,9 +88,6 @@ or
 |12000 | <img src="results/12000.png" width="200">|
 |14000 | <img src="results/14000.png" width="200">|
 
-### Other Implementations
-* [Magnet Loss in Tensorflow](https://github.com/pumpikano/tf-magnet-loss)
-
 ### Citing MagnetLoss-PyTorch
 If you use MagnetLoss-PyTorch in a scientific publication, I would appreciate references to the source code.
 

requirements.txt

@@ -0,0 +1,10 @@
+matplotlib
+numpy
+pandas
+pillow
+scipy
+seaborn
+setuptools
+sklearn
+tqdm
+visdom