forked from wffjwbbf/ComDesignProject
Readme Update
This commit is contained in:
parent
4bd11b77a8
commit
2fcdc581d8
Binary file not shown.
|
After Width: | Height: | Size: 434 KiB |
36
README.md
36
README.md
|
|
@ -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`
|
||||

|
||||
|
||||
主要是创建一个已加载权重的模型(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实现")
|
||||
|
|
|
|||
|
|
@ -1,3 +0,0 @@
|
|||
# Deep Sort
|
||||
|
||||
This is the implemention of deep sort with pytorch.
|
||||
BIN
result.png
BIN
result.png
Binary file not shown.
|
Before Width: | Height: | Size: 304 KiB |
Loading…
Reference in New Issue