microsoft / MMdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[FCN support] Error when converting voc-fcn16s/voc-fcn32s from caffe to IR

seanchung2 opened this issue · comments

Hi,

Thank you for fixing the other bugs.

Here is my model: fcn16s, fcn32s

Error message as following:

I1221 09:25:06.127544  6033 layer_factory.hpp:77] Creating layer data
F1221 09:25:06.127624  6033 layer_factory.hpp:81] Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: Python (known types: AbsVal, Accuracy, ArgMax, BNLL, BatchNorm, BatchReindex, Bias, Concat, ContrastiveLoss, Convolution, Crop, Data, Deconvolution, Dropout, DummyData, ELU, Eltwise, Embed, EuclideanLoss, Exp, Filter, Flatten, HDF5Data, HDF5Output, HingeLoss, Im2col, ImageData, InfogainLoss, InnerProduct, Input, LRN, LSTM, LSTMUnit, Log, MVN, MemoryData, MultinomialLogisticLoss, PReLU, Parameter, Pooling, Power, RNN, ReLU, Reduction, Reshape, SPP, Scale, Sigmoid, SigmoidCrossEntropyLoss, Silence, Slice, Softmax, SoftmaxWithLoss, Split, TanH, Threshold, Tile, WindowData)
*** Check failure stack trace: ***
Aborted (core dumped)

Thank you.

Hi @seanchung2 ,

Sorry for the late response. Newest code tested the conversion correctness from Caffe to TensorFlow.

Modify your prototxt to deploy.pbtxt

mmdownload -f caffe -n voc-fcn32s
mmconvert -sf caffe -in voc-fcn32s-deploy.prototxt -iw voc-fcn32s.caffemodel -df tensorflow -om ./caffe_fcn32s

Then you will get TensorFlow checkpoint files.

Thanks.