ppdet源码疑问
收藏
大佬们,我在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
收藏
请登录后评论