Skip to content

Commit c292bde

Browse files
committed
fix bugs of bot and byte
1 parent 23dbf40 commit c292bde

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414

1515
## 0. 近期更新
1616

17+
**2023.10.24**: 对于botsort(`tracker/botsort.py line 358`)和bytetrack(`tracker/bytetrack.py line 89`), 在低置信度检测列表初始化, 改正原有错误 (原本写错了, 写的用`det_high`初始化).
18+
1719
**2023.9.16**: 对于`track_demo.py`, 由于`yolov7``non_maximum_supress`函数中已经对类别做了筛选, 因此删去了主程序中类别计算的部分, 并修复了一些小bug
1820

1921
**2023.5.6[大更新]**: 对于v5, v7, 改变前处理和后处理方式(采用原有方式), ***解决了部分边界框近大远小的bug, 边界框更加精确***. 此外, 对于v8, 弃用了resize步骤, 直接推理.

tracker/botsort.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -355,7 +355,7 @@ def update(self, det_results, ori_img):
355355

356356
if det_low.shape[0] > 0:
357357
D_low = [STrack(cls, STrack.tlbr2tlwh(tlbr), score, kalman_format=self.opts.kalman_format)
358-
for (cls, tlbr, score) in zip(det_high[:, -1], det_high[:, :4], det_high[:, 4])]
358+
for (cls, tlbr, score) in zip(det_low[:, -1], det_low[:, :4], det_low[:, 4])]
359359
else:
360360
D_low = []
361361

tracker/bytetrack.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def update(self, det_results, ori_img):
8686

8787
if det_low.shape[0] > 0:
8888
D_low = [STrack(cls, STrack.tlbr2tlwh(tlbr), score, kalman_format=self.opts.kalman_format)
89-
for (cls, tlbr, score) in zip(det_high[:, -1], det_high[:, :4], det_high[:, 4])]
89+
for (cls, tlbr, score) in zip(det_low[:, -1], det_low[:, :4], det_low[:, 4])]
9090
else:
9191
D_low = []
9292

0 commit comments

Comments
 (0)