paddledetection infer_mot.py 预测mot/jde/jde_darknet53_30e_1088x608.yml 出现multiclass_nms3报错

[18325391772]

问题确认 Search before asking

• 我已经搜索过问题，但是没有找到解答。I have searched the question and found no related answer.

请提出你的问题 Please ask your question

使用官方模型预测mot/jde/jde_darknet53_30e_1088x608.yml报错

[liuhongen1234567]

您好，可以改一下 mot/jde/jde_darknet53_30e_1088x608.yml 配置文件中的 nms_top_k 和 keep_top_k， 把这两个改成一个小于231的值即可，比如都改成200. 官方示例中视频中的检测框可能比较多，所以才设置成2000. 我这边随便下了一个视频，然后修改后是没有问题的。

[liuhongen1234567]

这两个参数在配置文件的这个地方

PaddleDetection/configs/mot/jde/jde_darknet53_30e_1088x608.yml

Lines 42 to 45 in 7a4fc25

	keep_top_k: 500
	score_threshold: 0.01
	nms_threshold: 0.4
	nms_top_k: 2000

，我这边验证用的是这个视频 wget https://bj.bcebos.com/v1/paddledet/data/mot/demo/entrance_count_demo.mp4