Chinaunix首页 | 论坛 | 博客
  • 博客访问: 706069
  • 博文数量: 118
  • 博客积分: 1437
  • 博客等级: 上尉
  • 技术积分: 1155
  • 用 户 组: 普通用户
  • 注册时间: 2009-02-22 20:23
文章分类

全部博文(118)

文章存档

2022年(32)

2017年(3)

2014年(4)

2013年(1)

2011年(2)

2010年(16)

2009年(60)

我的朋友

分类: Python/Ruby

2022-03-25 17:31:11



deepsort包track.py执行的过程如下:
1. 执行track.py中的detect()函数
2. 实例化DeepSort类,执行deep_sort.py 中的初始化函数
3. 实例化FeatureExtractor,执行feature_extractor.py中的初始化函数__init__()
4. 建立模型,调用torchreid\models\__init__.py中的
build_model函数
5. 返回建立的模型的参数,调用models\下的模型定义文件(e.g. resnet .py),初始化模型结构
6. 调用torchreid\
utils\torchtools.py 文件中load_pretrained_weights,加载预训练权重
7. 返回track.py执行
deepsort.update()
8. 后续执行feature_extractor.py中的
__call__()函数,features = self.model(images)提取特征

包来源:mikel-brostrom Yolov5_DeepSort_Pytorch




阅读(883) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~