deepsort显示自己的label
(写了限定词 但是无论识别啥都是显示ID等问题)
今天有空,之前做的一个小项目用到了可达鸭大佬的项目中的代码,最近看到一些人问怎么在自己的项目中显示自己的标签,那么我今天也来斗胆分享下我的代码思路,希望对你们有帮助。
AIDetector_pytorch.py文件
- 首先更改这部分代码
def detect(self, im):
im0, img = self.preprocess(im)
pred = self.m(img, augment=False)[0]
pred = pred.float()
pred = non_max_suppression(pred, self.threshold, 0.4)
pred_boxes = []
for det in pred:
if det is not None and len(det):
det[:, :4] = scale_coords