History

L1aoXingyu acf363c181 1. Change loss function as a build-in attributes of heads 2. Update agw and bagtricks result		2020-03-16 15:23:09 +08:00
..
agwbaseline	1. Change loss function as a build-in attributes of heads	2020-03-16 15:23:09 +08:00
configs	1. Change loss function as a build-in attributes of heads	2020-03-16 15:23:09 +08:00
README.md	1. Change loss function as a build-in attributes of heads	2020-03-16 15:23:09 +08:00
train_net.py	1. Change loss function as a build-in attributes of heads	2020-03-16 15:23:09 +08:00

AGW Baseline in FastReID

Training

To train a model, run

CUDA_VISIBLE_DEVICES=gpus python train_net.py --config-file <config.yaml>

For example, to launch a end-to-end baseline training on market1501 dataset with ibn-net on 4 GPUs, one should excute:

CUDA_VISIBLE_DEVICES=0,1,2,3 python train_net.py --config-file='configs/AGW_market1501.yml'

Method	Pretrained	Rank@1	mAP	mINP
AGW	ImageNet	94.9%	87.4%	63.1%

Method	Pretrained	Rank@1	mAP	mINP
AGW	ImageNet	88.9%	79.1%	43.2%

Method	Pretrained	Rank@1	mAP	mINP
AGW	ImageNet	75.6%	52.6%	11.9%