def get_conv_filter(self, name): return tf.constant(self.data_dict[name][0], name="filter") def get_bias(self, name): return tf.constant(self.data_dict[name][1], name="biases") def get_fc_weight(self, name): return tf.constant(self.data_dict[name][0], name="weights")
时间: 2024-01-20 20:01:56 浏览: 54
arc_conv-master_sourcecode_PURE_game_arc_conv.exe_arc_conv_
5星 · 资源好评率100%
这段代码实现了获取预训练模型中某一层的权重参数和偏置参数的功能。其中,get_conv_filter函数获取卷积层的权重参数,get_bias函数获取偏置参数,get_fc_weight函数获取全连接层的权重参数。这些参数都是通过读取预训练模型文件(如VGGNet)中的参数来实现的,通过常量张量的形式返回。
阅读全文