PaddlePaddle / Anakin

High performance Cross-platform Inference-engine, you could run Anakin on x86-cpu,arm, nv-gpu, amd-gpu,bitmain and cambricon devices.
https://anakin.baidu.com/
Apache License 2.0
531 stars 135 forks source link

使用external_converter_v2目录下的converter.py转换caffemodle模型失败 #462

Open kezunlin opened 5 years ago

kezunlin commented 5 years ago

config.yaml

OPTIONS:
    Framework: CAFFE
    SavePath: ./output
    ResultName: ocr
    Config:
        LaunchBoard: ON
        Server:
            ip: 0.0.0.0
            port: 8888
        OptimizedGraph: 
            enable: OFF
            path: ./anakin_optimized/ocr.anakin.bin.saved
    LOGGER:
        LogToPath: ./log/
        WithColor: ON 

TARGET:
    CAFFE:
        # path to proto files
        ProtoPaths:
            - /home/kezunlin/program/caffe/src/caffe/proto/caffe.proto
        PrototxtPath: ./input_model/ocr/ocr.prototxt
        ModelPath: ./input_model/mnist/ocr.caffemodel

run converter and get error output:

$ python converter.py
INF | 2018-09-20 14:02:34.567382 | __init__.py:20 <module>() ]  Import Module: caffe_pb2
INF | 2018-09-20 14:02:34.581084 | parser_caffe.py:470 _Parsing_new() ]   [CAFFE] Parsing ...
Traceback (most recent call last):
  File "converter.py", line 20, in <module>
    graph = Graph(config)
  File "/home/kezunlin/program/anakin/tools/external_converter_v2/parser/graph.py", line 39, in __init__
    self.graph_io = self.parser()
  File "/home/kezunlin/program/anakin/tools/external_converter_v2/parser/caffe/parser_caffe.py", line 34, in __call__
    return self._Parsing_new()
  File "/home/kezunlin/program/anakin/tools/external_converter_v2/parser/caffe/parser_caffe.py", line 471, in _Parsing_new
    self._DetectionArch()
  File "/home/kezunlin/program/anakin/tools/external_converter_v2/parser/caffe/parser_caffe.py", line 41, in _DetectionArch
    self._FilterNet()
  File "/home/kezunlin/program/anakin/tools/external_converter_v2/parser/caffe/parser_caffe.py", line 389, in _FilterNet
    assert layer.name not in filtered_layer_names, " layer_name : %s" % (layer.name)
AssertionError:  layer_name : fusion5
Shixiaowei02 commented 5 years ago

您好,这是由于我们的 layer 目前还不支持 fusion5 导致的问题。请您将这一层的功能和你们的需求发邮件至 anakin@baidu.com,或者通过百度 hi 联系。谢谢!@kezunlin