Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support for bn_param in BatchNormalization #11

Open
vinaykumarngitub opened this issue Jan 17, 2020 · 2 comments
Open

Support for bn_param in BatchNormalization #11

vinaykumarngitub opened this issue Jan 17, 2020 · 2 comments

Comments

@vinaykumarngitub
Copy link

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
Copy link
Author

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
Copy link
Collaborator

ZQPei commented Jan 18, 2020

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants