From 72770f18688bb55e0a0eb5aaf389055eb3acdd6e Mon Sep 17 00:00:00 2001 From: Vedanuj Goswami Date: Fri, 7 Feb 2020 10:52:52 -0800 Subject: [PATCH] Change bbox to (x, y, width, height) --- script/extract_features_from_gt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)