XiaoMi / mace-models

Mobile AI Compute Engine Model Zoo
Apache License 2.0
371 stars 73 forks source link

Support object detection models #16

Closed XinCode closed 4 years ago

XinCode commented 5 years ago

Hi mace team, about object detection models: (1) ssd-mobilenetv1 only support caffe model, do you currently support tensorflow model? (2) Do you support other more advanced object detection models, like ssd with more complicated backbone, e.g., resnet50, or even faster rcnn models?

lydoc commented 5 years ago

Hi, (1) Yes, we do support the tensorflow model, you can use ClassPredictor and BoxEncodingPredictor as output nodes. (2) For faster rcnn models, there may be some ops we do not support now, but you can create a custom op if it is not supported yet. https://mace.readthedocs.io/en/latest/development/adding_a_new_op.html

XinCode commented 5 years ago

@lydoc noted with thanks.