Readme Update

This commit is contained in:
启程 2022-10-18 14:30:16 +08:00
parent 4bd11b77a8
commit 2fcdc581d8
5 changed files with 8 additions and 31 deletions

BIN
DeepSORT1.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 434 KiB

View File

@ -1,35 +1,15 @@
# 简化后的YOLOv7推断模块
# YOLOv7+DeepSORT行人检测与跟踪
## 程序余API介绍
## 简化后的YOLOv7推断模块
'主要的程序接口API全部位于detect.py和param.py两个文件中。'
***见yolov7-inferonly分支***
### 检测模块detect.py
## DeepSORT模块
主要是包含一个类和*一段示例代码*
0. `class YOLOv7(object)`:该类即为本程序提供的主要接口
* 效果图
1. `__init__(self,option:param.Parameters) -> None`
![avatar](DeepSORT1.png)
主要是创建一个已加载权重的模型model以供后续推断过程的使用之所以将函数与推断模型分开的原因是权重加载较慢这样做可以使得不必在每一次推断的过程中重复加载权重这对提高实验程序的实时性有很大帮助。尽管使用ONNX模型更加有利于提升速度但很多时候完成课程或毕业设计这样已经足够了
参数:`opt:param.Parameters`是位于param.py中的一个类仅仅是为了避免使用全局变量同时满足各种情形下的传参操作才这么设计。
返回值:无,但内部变量`self.model`是已加载参数的模型其父类为torch.nn.Moudules)。
## Reference
1. `detect(self,imgs:torch.Tensor) -> list`
参数:`imgs`(类型为torch.Tensor),大小应为(B,3,H,W),B为照片的张数当为单张照片时为B==1即可。3时RGB三通道W、H分别为图片的高和宽**要求为64的倍数**。
返回值:一个包含所有图片的bouding_box的列表列表中每个元素对应每一张图片其类型是二维的torch.Tensor最后一维长度为6分别对应(左上对角点横坐标`x1`,左上对角点纵坐标`y1`,右下角坐标横坐标`x2`,右下角纵坐标`y2`,可信度`conf`,类别`cls`),第二维是对应单张图片中的bouding_box。
### 传参模块param.py
主要是一个简单的类`Parameters`
| 变量名称 | 说明 |
| ------------ | ----------------------------------------------- |
| weights | 权重文件`.pt`文件的路径 |
| conf_thres | 置信度阈值,小于该置信度将被过滤 |
| iou_thres | 非极大值抑制(NMS)阈值,重叠比值超过该值将被合并 |
| device | 所部署到的设备`.to(device)` |
| classes | 选中的类,为`None`时代表全选 |
| agnostic_nms | NMS算法选项通常为`None`即可 |
| augment | 传入YOLOv7模型的参数通常为`None`即可 |
[DeepSORT实现-github](https://github.com/ZQPei/deep_sort_pytorch "DeepSORT实现")

View File

@ -1,3 +0,0 @@
# Deep Sort
This is the implemention of deep sort with pytorch.

Binary file not shown.

Before

Width:  |  Height:  |  Size: 304 KiB

BIN
test.png

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.7 MiB