Change bbox to (x, y, width, height)

facebookresearch · Feb 7, 2020 · 72770f1 · ishitamed19 · Apr 30, 2021 · 72770f1
1 parent 188e375
commit 72770f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)