首页 Paddle框架 帖子详情
roi-aligin 使用报错
收藏
快速回复
Paddle框架 问答模型训练深度学习 500 0
roi-aligin 使用报错
收藏
快速回复
Paddle框架 问答模型训练深度学习 500 0

ValueError: (InvalidArgument) roi_align(): argument 'X' (position 0) must be Tensor, but got Tensor (at ..\paddle\fluid\pybind\op_function_common.cc:737)

self.roi_align=RoIAlign(output_size=self.cfg.crop_size)# RoI Align
# features_multiscale      torch.Size([1, 1056, 87, 157])  这里是featuremap
# boxes_in_flat            torch.Size([12, 4])  # boxes_idx_flat           tensor([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0])
boxes_features=self.roi_align(features_multiscale,boxes_in_flat,boxes_idx_flat)
0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户