Ghustwb / MobileNet-SSD-TensorRT

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

Which are classes in your model? #16

Closed PiyalGeorge closed 5 years ago

PiyalGeorge commented 5 years ago

Hi @Ghustwb , I tested with your existing model in this repo, and its very accurate and almost perfect. Can you please tell me which are the classes in your existing model. i saw car and person and one must be background, but can you tell me which are the other classes?

I also would like to train a model as better and perfect as yours, can you tell me which dataset you used?, did you transfer learn or trained from scratch? and how many iterations did you go?

Ghustwb commented 5 years ago

person、car、bus、truck + background

PiyalGeorge commented 5 years ago

@Ghustwb Great thanks alot