Replicate MURA baseline and playing around

Mura competition https://stanfordmlgroup.github.io/competitions/mura/

Please Use Pull-Request If You Want To Update Master Branch, And Add sx5640 As Reviewer

How to run the script

1. Setup your preferred environment, i.e. virtualenv, conda, docker
2. Install packages: pip install -r requirements.txt
3. Download MURA dataset into /dataset/ directory. Link to MURA: https://stanfordmlgroup.github.io/competitions/mura/
4. Download MNIST dataset into /dataset/mnist/. Link to MNIST: http://yann.lecun.com/exdb/mnist/ Note: you need all 4 zip file and unpack them.
5. To run VGGNet model on MURA dataset, run python vggnet.py train. Add -h flag for all options.
6. To run a simple model on MNIST dataset, run python mnist_simple.py train. Add -h flag for all options.
7. To show a visualization of model attention or activation of a layer, run python visualize.py. Add -h flag for all options.

Note:

• In order to run tensorflow in gpu, install tensorflow-gpu instead of tensorflow in requirements.txt
• For MacOS user, importing dataset might throw exception, as explained by: https://matplotlib.org/faq/osx_framework.html If you use virtualenv, one solution pointed out by Kevin is to add the following code to the top of your dataset.py:

  import matplotlib as mpl
  mpl.use('TkAgg')