MMdnn icon indicating copy to clipboard operation
MMdnn copied to clipboard

Error when converting mxnet to pb

Open Zheweiqiu opened this issue 5 years ago • 3 comments

Platform (like ubuntu 16.04/win10): ubuntu 16.04 Python version: 3.7.3 Source framework with version (like Tensorflow 1.4.1 with GPU): mxnet 1.5.0 Destination framework with version (like CNTK 2.3 with GPU): pb Pre-trained model path (webpath or webdisk path): Pre-trained model available from here under "RetinaFace Pretrained Models" Running scripts: python -m mmdnn.conversion._script.convertToIR -f mxnet -n R50-symbol.json -w R50-0000.params -d resnet50 --inputShape 3,640,640 Got an warning: UserWarning: You created Module with Module(..., label_names=['softmax_label']) but input with name 'softmax_label' is not found in symbol.list_arguments(). Did you mean one of: data Size of three files I got: 20K resnet50.json 496K resnet50.npy 4.0K resnet50.pb

Have no idea how to fix this. Any help will be appreciate!

Zheweiqiu avatar Sep 21 '19 03:09 Zheweiqiu

I'm having the same issue. The original module has many more layers and weights than the generated module. @Zheweiqiu did you find a solution?

AlonSh avatar Feb 27 '20 12:02 AlonSh

I am converting pre-trained models of RetinaFace implemented by mxnet. I have the same problem. Did you fix this problem? @Zheweiqiu @AlonSh

cydawn avatar Jun 03 '20 15:06 cydawn

@cydawn I found a pytorch implementation, rewrote the model in keras and found a tool that loads pytorch params into tf graphs. I'm sorry but that work was done in my job using internal tools so I cannot share it /: Good luck mate

AlonSh avatar Jun 03 '20 16:06 AlonSh