update readme, adding pre-trained checkpoint, adding instruction for other datasets
Browse files
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
|