1. 安装依赖
pip install -r requirments.txt
2. Getting Started
运行
cd fastreid/evaluation/rank_cylib; make all
3. 下载好数据集
Getting_started,进行模型训练,这里需要修改文件地址:fasereid/data/dataset/market1501
添加数据集路径,Line30~Line32
dataset_dir = '/home/g303/lph/datasets/Market-1501-v15.09.15/'
dataset_url = 'http://188.138.127.15:81/Datasets/Market-1501-v15.09.15.zip'
dataset_name = "market1501"
运行命令即可训练。
python3 tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml MODEL.DEVICE "cuda:0"
然后使用tensorboard进行监控数据变化:
tensorboard --logdir logs/market1501/bagtricks_R50/
浏览器打开 http://localhost:6006/ 看效果
4. 评估
根据GETTING_STARTED.md里面的最后一条指令就可以进行evaluation。