[fbsync] Update coco_utils.py (#7869)

Summary: (Note: this ignores all push blocking failures!)

Reviewed By: matteobettini

Differential Revision: D48900402

fbshipit-source-id: eba6b527b7ae7b2db4634cc1ceec17cc1b218ba6

Co-authored-by: Nicolas Hug <nh.nicolas.hug@gmail.com>

references/detection/coco_utils.py

@@ -129,7 +129,7 @@ def convert_to_coco_api(ds):
         # find better way to get target
         # targets = ds.get_annotations(img_idx)
         img, targets = ds[img_idx]
-        image_id = targets["image_id"].item()
+        image_id = targets["image_id"]
         img_dict = {}
         img_dict["id"] = image_id
         img_dict["height"] = img.shape[-2]