From 87eb50b86fc0b8fd72989a0f6196792baef732b1 Mon Sep 17 00:00:00 2001
From: Ruizhe Cao <59573085+HyperbolicTangent@users.noreply.github.com>
Date: Sun, 31 Jan 2021 20:37:44 +0100
Subject: [PATCH] Update README.md

---
 README.md | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/README.md b/README.md
index af828ff..6339699 100644
--- a/README.md
+++ b/README.md
@@ -1,10 +1,11 @@
 # DenseNet121_DR
 Using transfer learning to classify diabetic retinopathy fundus images
 
-# Dataset
+# Run
+1. Dataset
 ```https://ieee-dataport.org/open-access/indian-diabetic-retinopathy-image-dataset-idrid```
 
-3. Dataset dir:
+2. Dataset dir:
 ```
 IDRID_Data/
     -train_image/
@@ -12,27 +13,22 @@ IDRID_Data/
     -train.csv
     -test.csv
 ```
-4. Prepare:\
+3. Prepare:\
 Change data_EDA directory. Run data_EDA to convert label, create TFRecord files, and show label distribution
 ```
 python input_pipeline/data_EDA.py
 ```
-5. Training and fine tuning:\
+4. Training and fine tuning:\
 Change dataset directory in config.gin, set train flag to Ture in main.py
 ```
 python main.py
 ```
-6. Evaluation:\
-Set train flag to False, change model directory to the saved model in log folder in main.py
-```
-python main.py
-```
-7. Hyperparameter tuning:\
+5. Hyperparameter tuning:\
 Change config.gin path to your own path
 ```
 python tune.py
 ```
-8. Deep Visualization:\
+6. Deep Visualization:\
 Change model_dir and dataset_dir to your own directory
 ```
 python grad_cam.py