功能更新列表

Introduction

本项目修改自mikel-brostrom的Yolov5_DeepSort_Pytorch主要针对船舶检测和追踪进行优化

安装pytorch 1.8.1版本
pip install -r requirements.txt
下载YOLOv5权重文件 https://github.com/ultralytics/yolov5/releases. 放在 yolov5/weights/目录下
下载deep_sort权重文件 https://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6. 放在deep_sort/deep/checkpoint/目录下；备份路径: https://pan.baidu.com/s/1b0WbHNk2ir2r6iEpeB665A 密码: makv
运行run_YoSort.py根据自己需要来修改启动参数
python3 run_YoSort.py --source rtsp://xxx(./image/xxx.jpg) --weights yolo_weight.pt --view-img(显示opencv检测画面)

Name		Name	Last commit message	Last commit date
Latest commit History 120 Commits
.idea		.idea
__pycache__		__pycache__
data		data
deep_sort_pytorch		deep_sort_pytorch
yolov5-old		yolov5-old
yolov5		yolov5
.DS_Store		.DS_Store
.gitattributes		.gitattributes
.gitignore		.gitignore
KalmanBox.py		KalmanBox.py
KalmanCV.py		KalmanCV.py
README.md		README.md
Town.gif		Town.gif
detect_YoSort.py		detect_YoSort.py
requirements-origin.txt		requirements-origin.txt
requirements.txt		requirements.txt
run_YoSort.py		run_YoSort.py
socket_function.py		socket_function.py
tools.py		tools.py
track.py		track.py