YOLOv5 + DeepSORT训练自己的数据

1. 配置环境

1.1 软硬件环境

X86 平台: Ubuntu20.04 with CUDA 11.0 and cuDNN 8.6.5
显卡型号：GTX1650
Nvidia驱动：473.15
嵌入式平台：NVIDIA Jetson Xavier NX
TensorRT 7.0.0.1
PyTorch 1.7.0_cu11.0 and TorchVision 0.8.1+cu110
OpenCV-Python 4.2
pycuda 2021.1

1.2 获取代码及资源

git clone --recurse-submodules https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch.git

重识别模型：
https://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6

2. 代码测试

python track.py --source test.mp4  --show-id

发现报错：Orderdicts has no " XXX"
原因：yolov5有版本，不能乱用，要从一而终。

3. 训练自己的数据

python train.py --img 640 --batch 4 --epoch 300 --data ./data/myvoc.yaml --cfg ./models/yolov5s --wegts/yolov5s.pt --workers 4

报错：no labels XXX
原因：dataset文件夹下把源图片文件夹名字默认成“images”了，如果你的对应文件夹不是“images”要改为“images”.