roi-aligin 使用报错
收藏
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
收藏
请登录后评论