コード例 #1
0
ファイル: tensorflow2caffe.py プロジェクト: yyqgood/bolt
 def add_concat(self, input_names, output_name, axis):
     layer = caffe_net.LayerParameter(name=output_name, type='Concat',
                 bottom=input_names, top=[output_name])
     layer.concat_param(axis)
     self.caffe_model.add_layer(layer)
     inputs = []
     for input_name in input_names:
         inputs.append(self.data_dict[input_name])
     self.data_dict[output_name] = Operators.concat(inputs, axis, output_name)
     return output_name