add Readme and setup

cw1204772 · Mar 16, 2018 · 5a6cc8b · 5a6cc8b
1 parent 2e8d83f
commit 5a6cc8b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/ReID/ReID_CNN/setup.sh b/ReID/ReID_CNN/setup.sh
@@ -1,2 +1,5 @@
 VeRi_dir=$1
+comp_dir=$2
+sv_dir=$3
 python3 create_VeRi_database.py --img_dir $VeRi_dir/image_train --query_dir $VeRi_dir/image_query --gallery_dir $VeRi_dir/image_test --label_dir $VeRi_dir/train_label.xml --train_txt VeRi_train_info.txt --query_txt VeRi_query_info.txt --gallery_txt VeRi_gallery_info.txt
+python3 create_Comp_database.py --sv_dataset_dir $sv_dir --comp_dataset_dir $comp_dir --info_dir Comp_info  
diff --git a/Readme.md b/Readme.md
@@ -34,7 +34,7 @@ cd ReID/ReID_CNN
 
 1. Create train, query, gallery info .txt files
    ```
-   bash setup.sh <dir_to_VeRi>
+   bash setup.sh <dir_to_VeRi> <dir_to_comp_dataset> <dir_to_sv_dataset>
    ```
 
 2. Training