MMdnn icon indicating copy to clipboard operation
MMdnn copied to clipboard

Convert trained yolov3 Darknet-53 custom model to tensorflow model

Open analyticalrohit opened this issue 5 years ago • 1 comments

Platform: ubuntu 16.04 Python version: 3.6 Source framework: mxnet (gluon-cv) Destination framework: Tensorflow Model Type: Object detection Pre-trained model path: https://gluon-cv.mxnet.io/build/examples_detection/demo_yolo.html#sphx-glr-build-examples-detection-demo-yolo-py

I have trained a custom yolov3 Darknet-53 model (yolo3_darknet53_custom.params) using gluon-cv (mxnet). I need to convert the yolo3_darknet53_custom.params (mxnet) model to yolo3_darknet53_custom.pb (tensorflow)

Also, I see https://pypi.org/project/mmdnn/ object detection is under on-going Models.

Queries: Does mmdnn supports yolo models (object detection) conversion trained using gluon-cv (mxnet) in general? Is there any way or work around by which I can convert yolo models?

Any leads would be great! Thank you

analyticalrohit avatar May 07 '20 14:05 analyticalrohit

Hi @rohit167 , we haven't test MMdnn with yolo models, so there might be some unsupported ops and accuracy loss.

XiaoXYe avatar May 08 '20 18:05 XiaoXYe