yhzhai commited on
Commit
2800337
·
1 Parent(s): 88677a1

update readme, adding pre-trained checkpoint, adding instruction for other datasets

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -28,6 +28,8 @@ pip install -r requirements.txt
28
  We provide preprocessed CASIA (v1 and v2), Columbia, and Coverage datasets [here](https://buffalo.box.com/s/2t3eqvwp7ua2ircpdx12sfq04sne4x50).
29
  Place them under the `data` folder.
30
 
 
 
31
 
32
  ## 3. Training and evaluation
33
 
@@ -37,6 +39,7 @@ Runing the following script to train on CASIAv2, and evalute on CASIAv1, Columbi
37
  python main.py --load configs/final.yaml
38
  ```
39
 
 
40
 
41
  ## Citation
42
  If you feel this project is helpful, please consider citing our paper
 
28
  We provide preprocessed CASIA (v1 and v2), Columbia, and Coverage datasets [here](https://buffalo.box.com/s/2t3eqvwp7ua2ircpdx12sfq04sne4x50).
29
  Place them under the `data` folder.
30
 
31
+ For other datasets, please prepare a json datalist file with similar structure as the existing datalist files in the `data` folder. After that, adjust the `train_dataslist` or the `val_datalist` entries in the configuration files `configs/final.yaml`.
32
+
33
 
34
  ## 3. Training and evaluation
35
 
 
39
  python main.py --load configs/final.yaml
40
  ```
41
 
42
+ Besides, our pre-trained checkpoint is available [here](https://buffalo.box.com/s/2t3eqvwp7ua2ircpdx12sfq04sne4x50).
43
 
44
  ## Citation
45
  If you feel this project is helpful, please consider citing our paper