CaptainEven / MCMOT

Real time one-stage multi-class & multi-object tracking based on anchor-free detection and ReID
MIT License
390 stars 83 forks source link

有的类只做检测 #39

Closed starsky68 closed 4 years ago

starsky68 commented 4 years ago

@CaptainEven 请问,对于有的类别制作检测的问题,我没有搞懂这个问题 如果已经训练了一个30个类别的模型,只对其中的0,4,6做跟踪其他的做检测,那么只需要在reid_cls_ids中设置0,4,6就可以吗,我这样设置的,然后现实这个错误: Fix size testing. training chunk_sizes: [2, 3, 3, 2] The output will be saved to /home/sunyue/MCMOT-master/src/lib/../../exp/mot/default13 Net input image size: 1088×608 [Err]: configuration conflict of reid_cls_ids and num_classes! Traceback (most recent call last): File "demo.py", line 219, in run_demo(opt) File "demo.py", line 41, in run_demo result_root = opt.output_root if opt.output_root != '' else '.' AttributeError: 'NoneType' object has no attribute 'output_root'

还是在训练的时候这么设置,在训练的时候设置也会出错。

CaptainEven commented 4 years ago

@starsky68 目前,同时做检测和跟踪,这里的确是有问题的,简单的解决办法是单独定一个参数num_classes = num_dets_classes + num_track_classes,然后就可以跟re_id_classes区分开来,当然,还需要修改相应的代码。