jackwoo0831 / yolov7-tracker

Yolo v5, v7, v8 and several Multi-Object Tracker(SORT, DeepSORT, ByteTrack, BoT-SORT, etc.) in MOT17 and VisDrone2019 Dataset. It uses a unified style and integrated tracker for easy embedding in your own projects.

• 👋 Hi, I’m @JackWoo0831
• 👀 I’m interested in Computer Vision and Multimodality.
• 🌱 My motto: 𝙏𝙤𝙤𝙠 𝙩𝙝𝙚 𝙨𝙤𝙪𝙧𝙚𝙨𝙩 𝙡𝙚𝙢𝙤𝙣 𝙩𝙝𝙖𝙩 𝙡𝙞𝙛𝙚 𝙝𝙖𝙨 𝙩𝙤 𝙤𝙛𝙛𝙚𝙧 𝙖𝙣𝙙 𝙩𝙪𝙧𝙣 𝙞𝙩 𝙞𝙣𝙩𝙤 𝙨𝙤𝙢𝙚𝙩𝙝𝙞𝙣𝙜 𝙧𝙚𝙨𝙚𝙢𝙗𝙡𝙞𝙣𝙜 𝙡𝙚𝙢𝙤𝙣𝙖𝙙𝙚.

yolov7-tracker's Issues

有C++实现吗

关于训练MOT17数据集报错：No labels in mot17/tarin.cache.

你好，我想请叫一下这个问题。我在训练mot17数据集的时候，按照readme中提到的方法训练时报AssertionError:train: No labels in mot17/train.cache.这样的错误，导致训练无法进行。后面提示时参考yolov5制作数据集，没找到解决方案。mot17数据集是直接下载官方的，我是先执行tools下面的covert_MOT17_to_yolo.py转换了格式（切了train、val、test三个），接着使用CUDA_VISIBLE_DEVICES=0 python train_aux.py --dataset mot17 --workers 8 --device 0 --batch-size 16 --data data/mot17.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights './weights/yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml命令试图进行训练，但是爆了上面的错误（No labels in mot17/train.cache.），希望帮忙解决，谢谢。

实验性能问题

您好，因为我不是**人，所以即使中文很奇怪也请谅解。 https://ieeexplore.ieee.org/document/9915399 不是别的，而是在Visdronedataset中通过将多个tracker与YOLOX结合的结果。 为什么你的代码和这个paper不同，SORT性能最好？ 而且与这个paper相比，整体性能较低，有什么理由呢？

MOT20使用yolo格式，运行tracker报错

11.27 pull 代码，执行报错
运行命令：

python tracker/track.py --dataset yolo --data_format yolo --tracker deepsort --model_path best.pt

报错如下：
yaml.scanner.ScannerError: sequence entries are not allowed here
in "./tracker/config_files/mot.yaml", line 24, column 21

能否列出您 mot 的 yaml 配置文件。

评估自己的数据集遇到的问题

我想问一下track eval是只能对行人进行评估吗？我自制的动物数据集在跟踪代码运行最后就会报错trackeval.utils.TrackEvalException: Attempted to evaluate an invalid class. Only pedestrian class is valid.
请问有没有其他的代码可以让我进行评估呢？

关于代码track_demo.py的错误

您好，我在使用track_demo.py脚本时，使用自己的视频数据，程序运行后报错如下：
Traceback (most recent call last):
File "tracker/track_demo.py", line 330, in
main(opts)
File "tracker/track_demo.py", line 154, in main
plot_img(img0, frame_id, [cur_tlwh, cur_id, cur_cls], save_dir=os.path.join(SAVE_FOLDER, 'reuslt_images', obj_name))
File "tracker/track_demo.py", line 252, in plot_img
text = f'{CATEGORY_DICT[cls]}-{id}'
KeyError: 3.0
请问是否为视频错误？

用yolov5自己训练的模型时运行track_demo.py时报错

您好，我自己对yolov5s.yaml做了改进，训练生成的模型放入track_demo.py时报错
Traceback (most recent call last):
File "E:/BaiduNetdiskDownload/yolov7-tracker-master/tracker/track_demo.py", line 330, in
main(opts)
File "E:/BaiduNetdiskDownload/yolov7-tracker-master/tracker/track_demo.py", line 117, in main
out = model(img.to(device)) # model forward
File "D:\Can\anac_install\envs\torch_can\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "E:\BaiduNetdiskDownload\yolov7-tracker-master\models\yolo.py", line 319, in forward
return self.forward_once(x, profile) # single-scale inference, train
File "E:\BaiduNetdiskDownload\yolov7-tracker-master\models\yolo.py", line 345, in forward_once
x = m(x) # run
File "D:\Can\anac_install\envs\torch_can\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "E:\BaiduNetdiskDownload\yolov7-tracker-master\models\yolo.py", line 54, in forward
y[..., 2:4] = (y[..., 2:4] * 2) ** 2 * self.anchor_grid[i] # wh
RuntimeError: The size of tensor a (84) must match the size of tensor b (48) at non-singleton dimension 2

Process finished with exit code 1
不知道问题在哪 想请教一下您 谢谢

请问您代码中的train.cache这样的cache文件是怎么生成的啊？

因为我在跑代码时，使用train_aux.py训练时发现train.cache和val.cache文件系统提示can't open/read file: check file path/integrity.
最后会导致报AssertionError: Image Not Found /data/wujiapeng/datasets/VisDrone2019/VisDrone2019/images/VisDrone2019-MOT-train/uav0000143_02250_v/0000001.jpg
为了保持路径一致，我特意建立了/data/wujiapeng/datasets/VisDrone2019文件夹去包含VisDrone2019数据集。

yolov5报错，生成的图片全部打不开

请教一些问题，感谢

您好，在查找多目标跟踪方案的时候看到了您的工作，感觉很棒很感兴趣，想学习一下然后试着用到自己的论文里。
还不太懂目标跟踪任务，有几个问题想请教一下，希望能够被解答，谢谢~
1.您是用yolov7官方提供的预训练模型，在VisDrone2019-MOT数据集上训练的吗？
2.这是只再训练了检测器yolov7呢，还是同时训练了跟踪器呢？比如bytetrack，这个跟踪器本身有需要训练学习的参数吗？
3.您提供的模型权重是只训练了10 epoch的吗？
4.weights文件夹下的DHN.pth和ckpt.t7两个权重文件是什么的呢？跟踪器的吗？

about StrongSORT

Looking forward to joining StrongSORT...
Best wishes

跟踪单个类别

大佬您好，想问一下如果想只跟踪一个类或者是只显示跟踪的一个类还应该修改哪里呀，visdrone检测模型有好几个类，如果修改为CATEGORY_NAMES = ['car']就会出现KeyError: 4.0这个问题，字典对不上，想问一下大佬，还应该修改哪里呀

MOTA评价指标的一点小问题

作者您好，我在使用您提供的在visdrone数据集上的预训练权重进行soft tracker的时候，测试出来的MOTA是负数，请问这是正常现象吗？希望作者空闲的时候能够回复我一下，万分感谢。

关于使用track_demo.py测试自定义视频的错误

使用您的代码测试自定义视频想保存txt格式的结果却保存不了。

环境问题

提示cython_bbox包没有bbox_overlaps，请问是因为我windows的问题还是因为包的版本问题，您的cython-bbox是什么版本呀

使用自己的视频进行跟踪测试

哈喽，想请教一下如何使用自己的视频进行跟踪测试。按照教程运行track.py文件时，更改了第35和81行DATA_ROOT的地址（注释上写这是数据集的目录），然后就会报错AssertionError: path must be your dataset path。点开是tracker_dataloader文件中的33行报错，于是更改了第29行DATA_ROOT的地址，还是有这个报错，就觉得挺奇怪的，使用跟踪应该不需要数据集吧，望解答，感谢感谢。

关于为bytetrack引入外观特征模块

作者您好呀，看到您在bytetrack的源码中加入了和deepsort一样的外观特征，原本的bytetrack源码似乎没有这部分。请问您加这一步是提升了最后的指标吗。我测试的好像加不加没有区别，感觉是我测试的有问题，所以来请教您，十分感谢~

关于评估自己的数据集的追踪效果

您好，可以请教一下您是怎么测出MOTA，IDS等评价指标的吗，请问我如果想利用自己的视频进行评估，计算MOTA等评价指标，是不是需要先手动对一段视频用darklabel进行标注，以mot的格式保存得到gt.txt。然后跑不同追踪器的推理源码，对同一段视频进行追踪，得到追踪结果的txt文件。最后去网上找相关代码，比较这两个txt文件，来计算MOTA之类的评价指标。请问我理解的对不对呀，我在网上查的有点凌乱，有人用github上发布的TrackEval测评价指标，有人用MOT2017数据集能直接上传到MOT challenge官网上进行测评，没有太查到说怎么测评自己数据集的。

autoanchor:Analyzing anchors...Killed

请问使用该代码按照您的流程训练visdrone数据集报错要怎么解决？谢谢

使用track_demo测试自定义视频错误

运行的时候，内存用完

大佬，您好！使用tracker_demo运行自己的单个视频时，卡死两次，第三次打开控制台，发现运行的时候内存暴涨，怎么调整batch_size之类的？

question about deepMOT

Hello, thank you for your perfect work. I have a question about the code that I want to ask you. There is a data set used to train DHN in deepmot -- DHN_ Data, how do you generate the. npy file in this dataset?

I would appreciate it if you could help me!

autoanchor:anchors...killed

tracker實用性

大大好
想請問你這幾個tracker，你覺得哪一個表現比較好啊
想聽聽你的看法

還有一個問題，如果要real time的話(如RTSP)，有比較推哪一個追蹤嗎?
謝謝

Request: OCSORT

Dear @JackWoo0831

I have come across your repo whilst looking for MOT implementations compatible with YOLOv7. You have done an amazing job so far, keep up with the work. It's great to see current MOT implementations for YOLOv7 at one place.

Will there be an implementation for OCSORT in the near future?

How is the ByteTrack implementation built? Is it based on OCSORT + Byte?

visdrone2019 Dataset是用的det还是mot部分啊？

据我所知，visdrone2019数据集有五种类型，det、vid、sot、mot、crowd，那么我想问在这个项目中作者你是用的哪个？不知道是det还是mot。

加值多目標計算功能

大大好
謝謝你提供的作品，很實用
但不知道能不能多開發幾個功能，例如這個影片
https://www.youtube.com/watch?v=XwzbFzqhF1Y
可以自訂各車道計算車流量、顯示車軌跡，時速的話，可以不急著弄，我覺得可能誤差會很大
如果你需要測試demo影片，我這邊可以提供讓你測試
最後謝謝你的偉大作品

路径报错

请问您可以增加MOT数据集的训练、验证、测试模块吗? 谢谢！

Discussion on small objects and long-tail dataset

感谢您的工作。
我想请教的问题刚好符合TO-DO list 的第二项，即类别不平衡及小目标问题。
就此，我想与您交流一下意见。

Error：size in bbox exist zero,（115,0,3）；--save_videos 运行结束没有video视频生成

运行tracker/track.py 中 deepsort 会遇到 uav0000077_00720_v:773it [01:46，8.24it/s]Error：size in bbox exist zero,（115,0,3）；还有一个问题，track.py 加 --save_videos 运行结束，没有video视频生成。请问作者有遇到吗？ 谢谢。

yolox

测试mot17路径问题

BUG: Appearance based features = True

Hello,

When I use appearance based features = True in bytetrack, detections in the next frame are made with coordinates of the previous frame.

Frame 0:

Frame 1:

All the bounding boxes shift to the right when camera moves to the right

有用c/c++重写的计划吗

工程看着很爽，简单明了，想问一下有改成c/c++的计划吗

YOLOv5采用哪个版本

camera track

可以调用摄像头实时跟踪吗？

FPS

你好，请问跟踪之后的结果，fps是指从检测到跟踪的fps吗？还是只是跟踪部分的fps？

RTSP or not

你好，请问支持读取RTSP吗，谢谢

训练自己数据集的一些问题

我是用yoloV5结合deepsort来进行多目标的跟踪，使用的数据集是自己的数据集。目前已经在yolo中训练完成，得到了对应我自己数据集的权重文件。但是deepsort直接调用训练完的yolo的权重进行预测的时候，目标物的id会随机变化。我想这应该是跟ReID有关系吧，因为我的数据集不是行人也不是车辆。所以我想请教你以下问题：
①我是不是应该对应我自己的数据集重新训练deepsort的ReID？
②如果我需要重新训练我自己的ReID，我应该如何制作数据集，格式是什么，如何分类？比如我跟踪的目标物都是一种类别，但是他们有很多个体，那么我的数据集中train和test都应该只有一个文件夹，然后文件夹下存放的是根据yolo标签识别后的不同目标个体裁切图像吗？
③我应该如何根据制作好的数据集重新训练我自己的ReID？
感谢你的回复

抽 pedestrain和people的部分帧训练，track.py追踪后有NaN跟负值的问题

请问yolov7-w6抽两类 pedestrain和people的部分帧图片训练，运行track.py追踪后有NaN跟负值的问题，怎么解决？

测试

你好，很感谢你的工作，让我看到了毕业的曙光！！！
下面是几个问题：1.如果我想直接用视频输入得到跟踪结果，应该做怎样的改正呢？2.另外，请问一下您使用的设备型号是什么呢，我使用的设备sort和deepsort的fps和你近似，但是uavmot几乎只有一半。（多次实验的结果）3.直接使用yolov5的tracker好像有一点问题，提示NameError: name 'DetectMultiBackend' is not defined，请问是有什么模块我没下载吗？

非常希望可以得到您的一些解答！！谢谢！！

TrackVideo

作者你好，咱们这个项目支持跟踪视频吗，看了一下代码好像只支持图片的跟踪，如果支持麻烦帮忙介绍一下，我也是个初学者有点搞不明白了，要是不支持视频的话那是否容易实现呢，毕竟跟踪大多数是跟踪视频的，转化成图片再输入确实不太方便，期待回复，十分感谢。

yolov8

作者你好，能出一个yolov8的版本吗？我尝试加了一下，但是结果是有问题的

Skip frame

tracker.update_without_detection(None, img0)
I found this line in traker.py file for skip frame.
but cant found function update_without_detection(None, img0) in any tracker

can you please how it work ?

数据集处理

您好！请问您对于VisDrone-MOT数据集中，不同视频图片大小不同是如何处理的呢？

作者您好，请问您是否有编写过一份Trackformer-bytetrack的项目呢

作者您好，请问您是否有编写过一份Trackformer-bytetrack的项目呢

About dataloader

现在我已经有针对自己的数据集使用yolov7训练好的模型，现在想直接加载我的模型和视频来进行跟踪任务。
所以请问一下可以增加数据输入为视频的选项吗？

Maybe a mistake in botsort.py

Yolov7-tracker/tracker/botsort.py

Line 447 in 421229c

track = strack_pool[idx]

I think it should be :track=u_tracks0[idx]

KeyError: 4.0

大佬你好，我在利用DeepSORT进行跟踪时，加入了--save_images进行图片保存，但是出现了报错，具体如下：
uav0000009_03358_v: 0it [00:00, ?it/s]Traceback (most recent call last):
File "tracker/track.py", line 339, in
main(opts)
File "tracker/track.py", line 179, in main
plot_img(img0, frame_id, [cur_tlwh, cur_id, cur_cls], save_dir=os.path.join(DATASET_ROOT, 'reuslt_images', seq))
File "tracker/track.py", line 252, in plot_img
text = f'{CATEGORY_DICT[cls]}-{id}'
KeyError: 4.0
能看一下是什么原因吗