diff --git a/DeepSORT1.png b/DeepSORT1.png new file mode 100644 index 0000000..660b83f Binary files /dev/null and b/DeepSORT1.png differ diff --git a/README.md b/README.md index 5203d19..ebf49f2 100644 --- a/README.md +++ b/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` +![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实现") diff --git a/deep_sort/README.md b/deep_sort/README.md deleted file mode 100644 index e89c9b3..0000000 --- a/deep_sort/README.md +++ /dev/null @@ -1,3 +0,0 @@ -# Deep Sort - -This is the implemention of deep sort with pytorch. \ No newline at end of file diff --git a/result.png b/result.png deleted file mode 100644 index b5b936f..0000000 Binary files a/result.png and /dev/null differ diff --git a/test.png b/test.png deleted file mode 100644 index 1dc4b0f..0000000 Binary files a/test.png and /dev/null differ