基于百度飞桨的单多人行人跟踪

基于百度飞桨的单/多人行人跟踪

代码参考：GitHub - PaddlePaddle/PaddleDetection at release/0.2

自己做了一些更改，下面是更改后的代码百度网盘链接：

链接：https://pan.baidu.com/s/10UBq2TRtFGORs30P9VaZxg
提取码：sdfo
复制这段内容后打开百度网盘手机App，操作更方便哦

一：百度飞桨的环境配置参考：飞桨PaddlePaddle-源于产业实践的开源深度学习平台

二：运行demo：

三：代码接口：

​ tools->run-interface.py

from tools.infer import entrance
import cv2
import datetime

if __name__ == '__main__':


    cap = cv2.VideoCapture('./1.mp4')  # 添加视频路径

    video_width = int(cap.get(3))
    video_height = int(cap.get(4))
    fps = int(cap.get(5)/2)
    print(fps)

    while True:
        _, im = cap.read()
        if im is None:
            break

        # d1 = datetime.datetime.now()
        cv2.imwrite("./0.jpg", im)
        # d2 = datetime.datetime.now()
        # print("读入时间差", (d2-d1).total_seconds())
        entrance("./0.jpg")
        img = cv2.imread(r"E:\code\PaddleDetection-release-0.2\contrib\PedestrianDetection\demo1\output\0.jpg")
        #raw = im.copy()
        # result = det.detect(im)
        cv2.imshow('a', img)


        if cv2.waitKey(1) == 27:
            break

    cap.release()
    cv2.destroyAllWindows()

配置好环境后，添加正确的视频路径即可正常运行：

cap = cv2.VideoCapture('./1.mp4')

的视频路径即可正常运行：

cap = cv2.VideoCapture('./1.mp4')