Ghustwb / MobileNet-SSD-TensorRT

Accelerate mobileNet-ssd with tensorRT
188 stars 91 forks source link

请问用自己训练的模型,类别不一样的时候要修改哪些地方? #30

Open wangkailovebaojiakun opened 5 years ago

wangkailovebaojiakun commented 5 years ago

请问要用自己训练的模型,类别数不一样,需要修改哪些地方呢?另外您说用5类做训练和检测,这个5类包含背景么?