wangjie-ruc / FGVC-CUB200

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FGVC-CUB200

training

具体设置可以按需修改

所有模型均未能复现原文结果,正在努力💪

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

models

  1. STN
  2. WS_DAN
  3. S3N

About


Languages

Language:Jupyter Notebook 97.8%Language:Python 2.2%