具体设置可以按需修改
所有模型均未能复现原文结果,正在努力💪
baseline
python train/resnnet50.py configs/resnet50.py --gpu 0 2
STN
python train/stn.py configs/stn.py --data_root /data/CUB_200_2011
S3N
python train/s3n.py configs/s3n.py --train_transforms configs/transforms/train2.json
WS-DAN
python train/wd_dan.py configs/ws_dan.py --work_dir out/WSDAN
- STN
- WS_DAN
- S3N