search

htshinichi / caffe-onnx

caffe model convert to onnx model
MIT License
175 stars 43 forks source link

Support for bn_param in BatchNormalization #11

Open vinaykumarngitub opened 4 years ago

vinaykumarngitub commented 4 years ago

We need support for the following parameters in BN layer: bn_param { scale_filler { type: "constant" value: 1 } shift_filler { type: "constant" value: 0.001 }

vinaykumarngitub commented 4 years ago

File "convert2onnx.py", line 34, in main(args) File "convert2onnx.py", line 26, in main graph, params = loadcaffemodel(caffe_graph_path,caffe_params_path) File "D:\VinayKumar\Programs\Task8(TRT_SegNet)\caffe-onnx-master\src\load_save_model.py", line 9, in loadcaffemodel text_format.Merge(open(net_path).read(), net) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 702, in Merge allow_unknown_field=allow_unknown_field) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 770, in MergeLines return parser.MergeLines(lines, message) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 795, in MergeLines self._ParseOrMerge(lines, message) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 817, in _ParseOrMerge self._MergeField(tokenizer, message) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 942, in _MergeField merger(tokenizer, message, field) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 1016, in _MergeMessageField self._MergeField(tokenizer, sub_message) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 942, in _MergeField merger(tokenizer, message, field) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 1016, in _MergeMessageField self._MergeField(tokenizer, sub_message) File "C:\ProgramData\Anaconda3\lib\site-packages\google\protobuf\text_format.py", line 909, in _MergeField (message_descriptor.full_name, name)) google.protobuf.text_format.ParseError: 58:2 : Message type "caffe.BatchNormParameter" has no field named "scale_filler"

ZQPei commented 4 years ago

The BatchNorm layer in your model is not the same as the official BN layer. I can not debug for you since I have no model of yours.