From c6da8afd01b55dc93c785eba5f04037ed0b900fd Mon Sep 17 00:00:00 2001 From: rvandeghen <37592623+rvandeghen@users.noreply.github.com> Date: Wed, 26 Jan 2022 18:26:57 +0100 Subject: [PATCH] Solve #5221 @datumbox --- references/detection/coco_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/references/detection/coco_utils.py b/references/detection/coco_utils.py index a6566028659..b0f193135ee 100644 --- a/references/detection/coco_utils.py +++ b/references/detection/coco_utils.py @@ -156,7 +156,7 @@ def convert_to_coco_api(ds): img_dict["height"] = img.shape[-2] img_dict["width"] = img.shape[-1] dataset["images"].append(img_dict) - bboxes = targets["boxes"] + bboxes = targets["boxes"].clone() bboxes[:, 2:] -= bboxes[:, :2] bboxes = bboxes.tolist() labels = targets["labels"].tolist()