diff --git a/script/extract_features_from_gt.py b/script/extract_features_from_gt.py index e653b5d..10e3f83 100644 --- a/script/extract_features_from_gt.py +++ b/script/extract_features_from_gt.py @@ -94,7 +94,7 @@ def get_batch_proposals(self, images, im_scales, im_infos, proposals): proposals_batch = [] for idx, img_info in enumerate(im_infos): boxes_tensor = torch.from_numpy( - proposals[idx]["bbox"][: int(proposals[idx]["num_box"]), 1:] + proposals[idx]["bbox"][: int(proposals[idx]["num_box"]), 0:] ).to("cuda") orig_image_size = (img_info["width"], img_info["height"]) boxes = BoxList(boxes_tensor, orig_image_size)