How to convert yolov3 model trained in caffe with self defined layer

XiaoMi / mace-models

Mobile AI Compute Engine Model Zoo

Apache License 2.0

371 stars 73 forks source link

How to convert yolov3 model trained in caffe with self defined layer #10

Closed lucasjinreal closed 5 years ago

lucasjinreal commented 5 years ago

How to convert own trained yolov3 model into mace ? Is there any convert tutorial?

llhe commented 5 years ago

For model conversion, check the documents: https://mace.readthedocs.io/en/latest/user_guide/advanced_usage.html

For yolo, you may need customized Ops: https://mace.readthedocs.io/en/latest/development/adding_a_new_op.html

gaopeipeiok commented 5 years ago

Did you complete yolo3 model on android。I have got the output datas，But I did not decode the datas，I am learning the yolo_layer.c. Maybe we could exchange experience .