首页 Paddle框架 帖子详情
roi-aligin 使用报错
收藏
快速回复
Paddle框架 问答模型训练深度学习 486 0
roi-aligin 使用报错
收藏
快速回复
Paddle框架 问答模型训练深度学习 486 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直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户