CaptainEven / FairMOTVehicle

A fork of FairMOT used to do vehicle MOT.用于跟踪车辆的多目标跟踪, 自定义数据进行单类别多目标实时跟踪
MIT License
185 stars 40 forks source link

How did you train the network? #10

Closed faruknane closed 4 years ago

faruknane commented 4 years ago

Hi, I'm really interested in fairmot work which is absolutely done perfectly. Could you please share a bit of the detail about how you trained the model for cars? Like did you freeze the body of the network and train only the heads? Did you train id head? How many images did you have for the training data? Did you encounter any generalization errors? I'll be so glad if you can answer my questions.

CaptainEven commented 4 years ago

Hi, thanks for interest. This repo is the extention work of FairMOT for one-class multi-object tracking, so the training details are almost the same as FairMOT except training dataset preparation. The content of training multi-vehicle using the DETRAC dataset is detailed in readme. In the training phase, no layers of the model are fixed, i.e all parameters are tuned during training process. The details of DETRAC dataset can be found in their official website. As for the generalization erros, from my point of view, it's good. If you are interested in multi-calss multi-object tracking as well, this is another extention work: https://github.com/CaptainEven/MCMOT

faruknane commented 4 years ago

@CaptainEven Thank you for the quick reply. I dig into MCMOT, there is a code below that draws gauss weight on the heatmap corresponding to that class id. I'm assuming, it writes 1 into some pixels that are close to the center of the object, right? image

CaptainEven commented 4 years ago

Yes