创建虚拟环境
conda create -n qdtrackenv python=3.8
conda activate your_env_name
安装软件包
pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
若运行出现No module named ‘mmcv._ext‘问题解决
下载mmcv使用的是pip install mmcv 指令,导致下载包不完整
在download.openmmlab.com/mmcv/dist/cu111/torch1.10/index.html该网站找到对应版本whl包
pip install mmcv_full-1.4.4-cp38-cp38-manylinux1_x86_64.whl即可成功安装
安装qdtrack模型
终端运行
python setup.py develop
若跑mot17数据集
以如下格式存放
生成注释
python tools/convert_datasets/mot2coco.py -i ./data/MOT17 -o data/MOT17/annotations --split-train --convert-det
训练模型
将qdtrack-frcnn_r50_fpn_4e_mot17.py文件里的路径进行修改
运行训练模型代码如下(将路径设置为该电脑下的路径)
python D:\****\qdtrack-master\tools/train.py D:\*****\qdtrack-master\configs\mot17\qdtrack-frcnn_r50_fpn_4e_mot17.py