コード例 #1
0
ファイル: tensorflow2caffe.py プロジェクト: yyqgood/bolt
 def add_l2norm(self, input_name, output_name):
     layer = caffe_net.LayerParameter(name=output_name, type='L2Norm',
                 bottom=[input_name],
                 top=[output_name])
     self.caffe_model.add_layer(layer)
     self.data_dict[output_name] = Operators.l2_norm(self.data_dict[input_name], output_name)
     return output_name