centernet _custom classes

dmlc / gluon-cv

Gluon CV Toolkit

http://gluon-cv.mxnet.io

Apache License 2.0

5.8k stars 1.21k forks source link

centernet _custom classes #1454

Open austinmw opened 3 years ago

austinmw commented 3 years ago

Hi, I noticed that the centernet detection model does not have a _custom version as the other detection models do.

How can I pass a custom classes list to get_model for centernet?

zhreshold commented 3 years ago

Yes, the custom versions of centernets are not available yet, as we are tight on the new features, anyone who can contribute the functionality is super welcome.