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
收藏
请登录后评论