update readme in projects/DML

pull/345/head
KaiyangZhou 2020-04-16 15:32:03 +01:00
parent f3df74026a
commit 2bbaa47ab3
2 changed files with 15 additions and 1 deletions

View File

@ -1 +1,14 @@
# Deep mutual learning
This repo implements [Deep Mutual Learning](https://arxiv.org/abs/1706.00384) (DML) for person re-id.
We used this code in our [OSNet](https://arxiv.org/pdf/1905.00953.pdf) paper (see Supp. B). The training command to reproduce the result of "triplet + DML" (Table 12f in the paper) is
```bash
python main.py \
--config-file im_osnet_x1_0_dml_256x128_amsgrad_cosine.yaml \
--root $DATA
```
Change `model.deploy` to `both` if you wanna enable model ensembling.
If you have any questions, please raise an issue in the Issues area.

View File

@ -1,6 +1,7 @@
model:
name: 'osnet_x1_0'
pretrained: True
deploy: 'model1'
data:
type: 'image'