Question

我在python tensorflow中使用slim库定义了一个网络模型。现在我想把它移植到c ++。但是在我看来，tensorflow.contrib.slim是一个只有python的库。我怎么能在我的c ++软件中使用它？

基本上我有类似的东西：

        self.conv1 = slim.convolution2d( \
            inputs=self.imageIn,num_outputs=32,\
            kernel_size=[8,8],stride=[4,4],padding='VALID', \
            biases_initializer=None,scope=myScope+'_conv1')

我怎样才能在c ++ tensorflow中实现它？

目前我正试图用简单的tensorflow替换slim实现，然后转向c ++。但是替换

self.conv1 = slim.convolution2d( \
            inputs=self.imageIn,num_outputs=32,\
            kernel_size=[8,8],stride=[1,4,4,1],padding='VALID', \
            biases_initializer=None, weights_initializer=_initializer, scope=myScope+'_conv1')

with plain tf

with tf.variable_scope(myScope+'_conv1'):
            weights = tf.get_variable("weights",[8, 8, 3, 32],
            initializer=_initializer,  dtype=tf.float32)

            self.conv1 = tf.nn.conv2d(self.imageIn, weights, [1, 4, 4, 1], padding='VALID')

在一个工作模型中产生一团糟：没有任何东西可以工作了。我忘记了什么？

非常感谢

在tensorflow c ++中导入slim模型

0 个答案: