首页 Paddle框架 帖子详情
ppdet源码疑问
收藏
快速回复
Paddle框架 问答学习资料 217 0
ppdet源码疑问
收藏
快速回复
Paddle框架 问答学习资料 217 0

大佬们,我在ppdet的源码里面看到这样的一行:

batch_size = paddle.slice(paddle.shape(im_shape), [0], [0], [1])

 

请问一下为什么要以这种方式获取batch_size?

如果是我我可能就会是

batch_size = paddle.to_tensor([im_shape.shape[0]])

学生党,问的问题可能比较浅薄,各位大佬见谅。

代码具体位置:

https://github.com/PaddlePaddle/PaddleDetection/blob/67767d2542e035d4e901c79eb790e6c4d89b35e5/ppdet/modeling/proposal_generator/rpn_head.py#L190

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户