bytetrack官方代码训练、测试、导出onnx

演示结果

gpu环境

CUDA 11.1
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c conda-forge

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
设置搜索时显示通道地址
conda config --set show_channel_urls yes
验证
conda config --get channels

conda create -n bytetrack python=3.8

conda activate bytetrack

pip uninstall pillow
pip install pillow==6.2.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/

conda install -c conda-forge lap

pip install -r E:\code\ByteTrack-main\requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

安装过程问题

• GeForce RTX 3060 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
  说明GPU与当前PyTorch安装不兼容的问题，原因是CUDA和cuDNN版本不适配
• 安装cython-bbox 需要Microsoft Visual C++ 14.0
• from yolox import _C 问题，cd到代码下安装python3 setup.py develop
• 运行测试代码 ValueError: numpy.ndarray has the wrong size, try recompiling. Expected 80, got 96 pip install numpy==1.23.4

训练

yolox的预训练模型放在<ByteTrack_HOME>/pretrained

python3 tools/train.py -f exps/example/mot/yolox_s_mix_det.py -d 1 -b 8 --fp16 -o -c pretrained/yolox_s.pth

yolox_s_mix_det.py 修改训练和验证的 data_dir名称和name，对应MOTDataset的data_dir也要改

测试

demo
python3 tools/demo_track.py video -f exps/example/mot/yolox_s_mix_det.py -c pretrained/bytetrack_s_mot17.pth.tar --fp16 --fuse --save_result

导出onnx

cd <ByteTrack_HOME>
python3 tools/export_onnx.py --output-name bytetrack_s.onnx -f exps/example/mot/yolox_s_mix_det.py -c pretrained/bytetrack_s_mot17.pth.tar