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”.