首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >将roi_pooling图层的输出运行到conv2d

将roi_pooling图层的输出运行到conv2d
EN

Stack Overflow用户
提问于 2019-11-11 16:00:20
回答 1查看 29关注 0票数 0

我已经从这个library实现了roi_pooling。

将roi_pooling作为运行后

代码语言:javascript
运行
复制
conv5_3 = net.end_points['conv5_3']
#implement ROI Pooling
input_boxes=tf.dtypes.cast(input_boxes,tf.int32)
pooled_features = roi_pooling(conv5_3, input_boxes, pool_height=5, pool_width=30)

那么pooled_features的形状是未知的。

代码语言:javascript
运行
复制
(Pdb) p pooled_features
<tf.Tensor 'RoiPooling:0' shape=<unknown> dtype=float32>

在roi_pooling之后,我的网络仍然需要通过另一个识别网。

识别网络是

代码语言:javascript
运行
复制
def recognitionnet(inputs, fatness = 64, dilation = True):
    """
    backbone net of vgg16
    """
    # End_points collect relevant activations for external use.
    end_points = {}
    # Original VGG-16 blocks.
    with slim.arg_scope([slim.conv2d, slim.max_pool2d], padding='SAME'):
        # Block1
        net = slim.repeat(inputs, 2, slim.conv2d, fatness, [3, 3], scope='conv1')
        end_points['conv1_2'] = net
        net = slim.max_pool2d(net, [2, 2], scope='pool1')
        end_points['pool1'] = net
        # Block 2.
        net = slim.repeat(net, 2, slim.conv2d, fatness * 2, [3, 3], scope='conv2')
        end_points['conv2_2'] = net
        net = slim.max_pool2d(net, [2, 2], scope='pool2')
        end_points['pool2'] = net
        # fc7 as conv
        net = slim.conv2d(net, fatness * 16, [1, 1], scope='fc7')
        end_points['fc7'] = net
        #model_summary()
        #from keras.utils.visualize_util import plot
        #plot(model, to_file='model.png')
    return net, end_points;  

但是当pooled_features通过识别网的第一层时,我会出错,因为

代码语言:javascript
运行
复制
TypeError: TypeErro...neType',)

当pooled_features通过net = slim.repeat(inputs, 2, slim.conv2d, fatness, [3, 3], scope='conv1')时,可能会出什么问题。

我该如何解决这个问题?

EN

回答 1

Stack Overflow用户

发布于 2019-11-21 15:33:07

我所做的是将这两个图分开。第一个图有roi_poling实现,第二个图有recognitionnet。

由于识别网位于单独的图形中,因此我们可以使用placeholder作为识别网图形的输入。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/58797512

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档