5ce43185b1
|
||
|---|---|---|
| .. | ||
| README.md | ||
| default_config.py | ||
| dml.py | ||
| im_osnet_x1_0_dml_256x128_amsgrad_cosine.yaml | ||
| main.py | ||
README.md
Deep mutual learning
This repo implements Deep Mutual Learning (CVPR'18) (DML) for person re-id.
We used this code in our OSNet paper (see Supp. B). The training command to reproduce the result of "triplet + DML" (Table 12f in the paper) is
python main.py \
--config-file im_osnet_x1_0_dml_256x128_amsgrad_cosine.yaml \
--root $DATA
$DATA corresponds to the path to your dataset folder.
Change model.deploy to both if you wanna enable model ensembling.
If you have any questions, please raise an issue in the Issues area.