Update README.md

README.md

@@ -2,20 +2,30 @@ Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for T
 ======
 This is a reimplementation of paper : Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for Thorax Disease Classification (AG-CNN)
 
-*Recently, the paper is accpeted in PRL 2020 with title "Thorax disease classification with attention guided convolutional neural network"*
+*Recently, the paper was accpeted in PRL 2020 with title "Thorax disease classification with attention guided convolutional neural network"*
 
 In this repo, you can train & test your own models, more details are shown in the code.
 
-method
+AG-CNN Method
 ------
 ![method](https://github.com/Ien001/AG-CNN/blob/master/Screen%20Shot%202019-04-03%20at%2011.45.38%20AM.png)
 
-result
+Result
 ------
 ![result](https://github.com/Ien001/AG-CNN/blob/master/result.png)
 
-model
+Model
 -----
 You can also try my models here:
-Google Drive: https://drive.google.com/drive/folders/1WREzoYmwzw0gOpfobty3quySJaTDAXUi?usp=sharing
-Baidu Cloud: https://pan.baidu.com/s/15Juub8f7gaBN826QuGGo8Q
+
+Google Drive: 
+https://drive.google.com/drive/folders/1WREzoYmwzw0gOpfobty3quySJaTDAXUi?usp=sharing
+
+Baidu Cloud: 
+https://pan.baidu.com/s/15Juub8f7gaBN826QuGGo8Q
+
+Docker
+-----
+You can use my docker container to avoid version issues:
+
+https://hub.docker.com/r/ianren/cuda9.0-cudnn7.0-tf1.5-keras2.1-pytorch1.0?utm_source=docker4mac_2.2.0.0&utm_medium=repo_open&utm_campaign=referral