output roi all zeros

[qchenclaire]

Hi I tried to test ROIAlign on images with rois = roi_align(detections, boxes, box_index) detections' shape is torch.Size([1, 3, 271, 271]) and boxes looks like tensor([[ 151.6779, 18.8237, 270.0000, 84.2876], [ 175.6971, 9.2199, 255.9987, 92.7847], [ 165.4188, 0.0000, 233.8400, 119.7061], [ 134.8676, 25.9375, 270.0000, 79.1737]], device='cuda:0') and box_index looks like tensor([ 0, 0, 0, 0], dtype=torch.int8, device='cuda:0') The output shape is (4, 50, 50, 3), which means 4 cropped images. But I got only the first image that looks correct and the rest 3 are all zeros.