ytimed2020 / pp-tracking_gui Goto Github PK

View Code? Open in Web Editor NEW

87.0 4.0 20.0 1.92 GB

Object detection GUI based on PaddleDetection(基于PaddleDetection的目标检测GUI框架）

Python 80.06% Shell 0.25% CMake 1.00% C++ 18.51% Makefile 0.17%

pp-tracking_gui's Introduction

PP-Tracking GUI界面测试版

本项目是基于飞桨开源的实时跟踪系统PP-Tracking开发的可视化界面

在PaddlePaddle中加入pyqt进行GUI页面研发，可使得整个训练过程可视化，并通过GUI界面进行调参，模型预测，视频输出等，通过多种类型的识别，简化整体预测流程。

GUI界面基于PyQT和PP-Tracking python部署代码开发；当前覆盖单镜头的全部功能，如行人跟踪，车辆跟踪，流量统计等

推荐使用Windows环境

主要包含两个步骤：

导入训练模型，修改模型名称
安装必要的依赖库
启动前端界面

1. 下载预测模型

PP-Tracking 提供了覆盖多种场景的预测模型，用户可以根据自己的实际使用场景在链接中直接下载表格最后一列的预测部署模型

如果您想自己训练得到更符合您场景需求的模型，可以参考快速开始文档训练并导出预测模型

模型导出放在./output_inference目录下

2. 必要的依赖库安装

pip install -r requirements.txt

其中包括以下依赖

pyqt5
moviepy
opencv-python
PySide2
matplotlib
scipy
Cython
cython_bbox
paddlepaddle
pycocotools
lap
sklearn
motmetrics
openpyxl

注：

Windows环境下，需要手动下载安装cython_bbox，然后将setup.py中的找到steup.py, 修改extra_compile_args=[’-Wno-cpp’]，替换为extra_compile_args = {'gcc': ['/Qstd=c99']}, 然后运行python setup.py build_ext install
numpy版本需要大于1.20

3. 启动前端界面

执行python main.py启动前端界面

参数说明如下:

参数	是否必须	含义
模型运行	Option	点击后进行模型训练
结果显示	Option	在运行状态为检测完成的时候进行结果视频显示
停止运行	Option	停止整个视频输出
取消轨迹	Option	在一开始时取消轨迹
阈值调试	Option	预测得分的阈值，默认为0.5
输入FPS	Option	输入视频的FPS
检测用时	Option	视频的检测时间
人流量检测	Option	每隔一段帧数内的人流量统计图表
时间长度	Option	人流量时间统计长度
开启出入口	Option	导入视频后可自行选择是否开启出入口训练
导出文件	Option	可视化结果保存的根目录，默认为output/

说明：

如果安装的PaddlePaddle不支持基于TensorRT进行预测，需要自行编译，详细可参考预测库编译教程。
建议使用windows环境进行运行

pp-tracking_gui's People

Contributors

Stargazers

Watchers

Forkers

yixinkristy jerrywgz liasazhang nemonameless qiuweibin2005 myuanz liansshang linxiao0915 jeck-liu-create july1921 fangtang6 hey-jobs northboy-su pipi0816 hsaigroup can1214

pp-tracking_gui's Issues

MTMCT可以用吗?为什么我运行之后会报错

报错如下:
AttributeError: 'PySide2.QtWidgets.QMainWindow' object has no attribute 'label_28'

目前gui界面起来啦实际功能很多报错
lzclzclzclzclzclzc
当前是行人模型
Traceback (most recent call last):
File "deploy/pptracking/python/mot_jde_infer.py", line 27, in
from det_infer import Detector, get_test_images, print_arguments, PredictConfig

冰山一角

无模型推理结果

请问模型预测后视频检测不出来人可能是什么原因？（已下载好预测模型并且代码无报错）

在使用gui检测小目标的车辆检测时，报错mot_jde_infer.py: error: unrecognized arguments: .mp4，

在使用gui进行车辆跟踪的小目标跟踪时，报错mot_jde_infer.py: error: unrecognized arguments: .mp4，
预测部署文件使用的是：fairmot_hrnetv2_w18_dlafpn_30e_576x320_visdrone_vehicle
视频文件是：https://bj.bcebos.com/v1/paddledet/data/mot/demo/visdrone_demo.mp4

No module named 'paddle.inference'

Traceback (most recent call last):
File "deploy/pptracking/python/mot_jde_infer.py", line 23, in
from paddle.inference import Config
ModuleNotFoundError: No module named 'paddle.inference'
Exception in thread Thread-1:
Traceback (most recent call last):
File "/home/deploy/anaconda3/envs/PP-Tracking_GUi/lib/python3.8/threading.py", line 932, in _bootstrap_inner
self.run()
File "/home/deploy/anaconda3/envs/PP-Tracking_GUi/lib/python3.8/threading.py", line 870, in run
self._target(*self._args, **self._kwargs)
File "main.py", line 571, in load_model
self.read_txt_file()
File "main.py", line 693, in read_txt_file
f = open('output/' + self.end_file_name + '_flow_statistic.txt', 'r')
FileNotFoundError: [Errno 2] 没有那个文件或目录: 'output/gd9_flow_statistic.txt'

请问在linux ubuntu系统运行GUI能够正常使用吗，为什么我无法显示gui界面呢？

运行日志：QFactoryLoader::QFactoryLoader() checking directory path "/usr/libexec/iconengines" ...
Unknown property box-shadow
Unknown property box-shadow

Unknown property box-shadow
Unknown property box-shadow
QFactoryLoader::QFactoryLoader() checking directory path "/gpfs1/home/kaizhuang7/.local/lib/python3.6/site-packages/PySide2/Qt/plugins/accessible" ...
QFactoryLoader::QFactoryLoader() checking directory path "/usr/libexec/accessible" ...

0526e81 一大堆已经删除的图片
2392ecb 一大堆已经删除的图片
5184309 一大堆已经删除的图片
一大堆已经删除的推理模型
一些奇怪的合并. 这个可以在合并的时候用 rebase 模式来避免.

我试着删了删没用的历史, 留下的 .git 还有 88M, 如果你觉得可以, 我就开一个 PR 出来.