From ff08404aaaa07a0ddcec7f35dc847baa29aad304 Mon Sep 17 00:00:00 2001 From: tino926 Date: Thu, 29 Jun 2023 17:33:15 +0800 Subject: [PATCH] remove objects with iscrowd=True in Objects365 Signed-off-by: tino926 --- data/Objects365.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/Objects365.yaml b/data/Objects365.yaml index 990f5a56b39e..d4045e2f852c 100644 --- a/data/Objects365.yaml +++ b/data/Objects365.yaml @@ -428,7 +428,7 @@ download: | path = Path(im["file_name"]) # image filename try: with open(labels / path.with_suffix('.txt').name, 'a') as file: - annIds = coco.getAnnIds(imgIds=im["id"], catIds=catIds, iscrowd=None) + annIds = coco.getAnnIds(imgIds=im["id"], catIds=catIds, iscrowd=False) for a in coco.loadAnns(annIds): x, y, w, h = a['bbox'] # bounding box in xywh (xy top-left corner) xyxy = np.array([x, y, x + w, y + h])[None] # pixels(1,4)