【目标跟踪】pytorch环境下的deep_sort_yolov3代码实现

女爷i 2023-07-17 14:16 5阅读 0赞

# 1 环境配置 #

## 1.1 源码下载 ##

[Github：deep\_sort\_pytorch][Github_deep_sort_pytorch]

**1 下载YOLOv3参数**  
[https://pjreddie.com/media/files/yolov3.weights][https_pjreddie.com_media_files_yolov3.weights]  
[https://pjreddie.com/media/files/yolov3-tiny.weights][https_pjreddie.com_media_files_yolov3-tiny.weights]  
放在`detector/YOLOv3/weight/`目录下。

**2 下载deepsort参数：**  
在[谷歌云盘][Link 1]或[百度云(密码：fbuw)][fbuw]中下载ckpt.t7，放入`deep_sort/deep/checkpoint/`目录下。

## 1.2 依赖 ##

系统：Window10  
IDE：Pycharm+Anaconda

在Anaconda Prompt内使用pip install安装下面的依赖：

Python3.6
    numpy
    scipy
    opencv-python
    sklearn
    vizer
    edict
    pillow-6.2.2
    pyyaml
    easydict

> 安装依赖时，如果下载过慢可以使用清华镜像，像这样：`pip install 名称 -i https://pypi.tuna.tsinghua.edu.cn/simple`

> pillow不能使用最新版本，最新版本缺少`PILLOW_VERSON`模块，将导致报错，所以我使用pillow-6.2.2版本。

在[Pytorch官网][Pytorch]下载所需版本的Pytorch和torchvision。下面是我选择的版本。

PyTorch-1.1.0
    torchvision-0.3.0

# 2 代码实现 #

使用终端进入项目目录下，输入命令`python yolov3_deepsort.py VIDEO_PATH`:![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1NlZV9TdGFy_size_16_color_FFFFFF_t_70]  
加载完成后，即可开始检测，放一张截图  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1NlZV9TdGFy_size_16_color_FFFFFF_t_70 1]  
用我自己的笔记本进行测试，用cpu测试时的FPS=0.5，用gpu测试时的FPS=3。这检测速度实在是太慢了。

程序能够运行起来了，接下来就是要去想办法修改参数、训练自己的数据集了。

第二次更新:  
使用终端运行太麻烦了，对`yolov3_deepsort.py`文件内的`main`进行修改，VIDEO\_PATH为视频的路径。

if __name__ == "__main__":
        VIDEO_PATH = "E:/Activities/fwwb2019/code/keras-yolo3-master/test/video2.mp4"
    
        parser = argparse.ArgumentParser()
        parser.add_argument("--VIDEO_PATH", type=str, default=VIDEO_PATH)
        parser.add_argument("--config_detection", type=str, default="./configs/yolov3.yaml")
        parser.add_argument("--config_deepsort", type=str, default="./configs/deep_sort.yaml")
        parser.add_argument("--ignore_display", dest="display", action="store_false", default=True)
        parser.add_argument("--frame_interval", type=int, default=1)
        parser.add_argument("--display_width", type=int, default=800)
        parser.add_argument("--display_height", type=int, default=600)
        parser.add_argument("--save_path", type=str, default="./demo/demo.avi")
        parser.add_argument("--cpu", dest="use_cuda", action="store_false", default=True)
        parser.add_argument("--camera", action="store", dest="cam", type=int, default="-1")
    
        args = parser.parse_args()
        cfg = get_config()
        cfg.merge_from_file(args.config_detection)
        cfg.merge_from_file(args.config_deepsort)
    
        with VideoTracker(cfg, args) as vdo_trk:
            vdo_trk.run()

[Github_deep_sort_pytorch]: https://github.com/ZQPei/deep_sort_pytorch
[https_pjreddie.com_media_files_yolov3.weights]: https://pjreddie.com/media/files/yolov3.weights
[https_pjreddie.com_media_files_yolov3-tiny.weights]: https://pjreddie.com/media/files/yolov3-tiny.weights
[Link 1]: https://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6
[fbuw]: https://pan.baidu.com/s/1YJ1iPpdFTlUyLFoonYvozg#list/path=/
[Pytorch]: https://pytorch.org/get-started/previous-versions/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1NlZV9TdGFy_size_16_color_FFFFFF_t_70]: /images/20230528/4884385c8dfa44f597b5de913ef5fb1b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1NlZV9TdGFy_size_16_color_FFFFFF_t_70 1]: /images/20230528/a2355d0d6f1944cabbf34e83e5805912.png