yolov5+deepsort用於目標跟蹤

從github下載安裝包

https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch

 

新建環境

cd到項目下

第一步，安裝依賴

安裝依賴

pip install -r requirements.txt

依賴安裝完成檢查自己的GPU是否可用

In [1]: import torch

In [2]: torch.cuda.current_device()

Out[2]: 0

In [3]: torch.cuda.device(0)

Out[3]: <torch.cuda.device at 0x7efce0b03be0>

In [4]: torch.cuda.device_count()

Out[4]: 1

In [5]: torch.cuda.get_device_name(0)

Out[5]: 'GeForce GTX 950M'

In [6]: torch.cuda.is_available()

Out[6]: True

可以檢測到就行。如果檢測不到GPU，可能是安裝的cpu版本的pytorch..這時候需要手動安裝pytorch的GPU版本

官網https://pytorch.org/get-started/locally/

 

 

 

復制下面的安裝命令行就行。找適合自己設備的cuda版本

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

 

 第二步，運行項目

從github 下載https://github.com/wudashuo/yolov5壓縮包，

解壓放在項目的yolov5文件夾下面，運行

python track.py --source ... --show-vid  # show live inference results as well

Video: --source file.mp4      # python track.py --source="011.mp4" --show-vid
Webcam: --source 0
RTSP stream: --source rtsp://170.93.143.139/rtplive/470011e600ef003a004ee33696235daa
HTTP stream: --source http://wmccpinetop.axiscam.net/mjpg/video.mjpg

至此，項目完成。source 0表示攝像頭。

 

PS這個過程走了很多彎路，原因是用了另一位博主的yolov5文件夾里的東西，模型不匹配，最后搞了一下午終於發現用原官網的就可以解決。

有問題歡迎留言區討論。