【歩行者の再識別】TransReIDの再発（例としてmarket1501）

コードリファレンス：https：//github.com/heshuting555/TransReID

ペーパーリファレンス：https：//arxiv.org/abs/2102.04378

1.環境

ubuntu16.04
python3.6
cuda110

torch==1.7.0+cu110
torchvision==0.8.0

timm
yacs
opencv-python==4.1.0.25

2.モデルの準備

https://www.kaggle.com/abhinand05/vit-base-models-pretrained-pytorch

3.変更

次のパスを独自のパスに変更します。

1）事前トレーニングモデルパス

2）データパス

3）モデルの保存パス

configs / Market / vit_transreid.ymlファイルを変更し、独自のファイルに変更します（最後のファイルは、モデルがTransReIDの上位ディレクトリに保存されているためです）。

MODEL:  
  PRETRAIN_PATH: '/home/***/.cache/torch/checkpoints/jx_vit_base_p16_224-80ecf9dd.pth'

DATASETS:
  NAMES: ('market1501')
  ROOT_DIR: ('/root/datasets/')

TEST:
  EVAL: True
  IMS_PER_BATCH: 256
  RE_RANKING: False
  WEIGHT: './logs/market_vit_transreid/transformer_120.pth'

OUTPUT_DIR: './logs/market_vit_transreid'

4.トレーニング

python train.py --config_file configs/Market/vit_transreid.yml MODEL.DEVICE_ID "('0')"