From a19815815d978f65871d15ec89b0ce6f1238d6a2 Mon Sep 17 00:00:00 2001
From: Hyunjong Park <hyunpark@yonsei.ac.kr>
Date: Fri, 28 Aug 2020 12:06:21 +0900
Subject: [PATCH] Update README.md

---
 README.md | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 463c3283..dd2ea327 100644
--- a/README.md
+++ b/README.md
@@ -39,11 +39,12 @@ python Train.py --gpus 1 --dataset_path 'your_dataset_directory' --dataset_type
 ## Evaluation
 * Test the model with our pre-trained model and memory items
 ```bash
-python Evaluate.py --model_dir pretrained_model.pth --m_items_dir m_items.pt
+python Evaluate.py --dataset_type ped2 --model_dir pretrained_model.pth --m_items_dir m_items.pt
 ```
 * Test your own model
+* Check your dataset_type (ped2, avenue or shanghai)
 ```bash
-python Evaluate.py --model_dir your_model.pth --m_items_dir your_m_items.pt
+python Evaluate.py --dataset_type ped2 --model_dir your_model.pth --m_items_dir your_m_items.pt
 ```
 
 ## Bibtex