From 9b377c40ea3010936493d0a56d9d597b30216cbc Mon Sep 17 00:00:00 2001 From: Josh Rice Date: Fri, 23 Feb 2024 06:42:41 -0600 Subject: [PATCH] check annotation category not cat collection --- cocojson/tools/merge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cocojson/tools/merge.py b/cocojson/tools/merge.py index 8f9bc40..8045777 100644 --- a/cocojson/tools/merge.py +++ b/cocojson/tools/merge.py @@ -84,7 +84,7 @@ def merge(jsons, img_roots, output_dir, cids=None, outname="merged"): merged_dict["images"].append(img) for annot in cocodict["annotations"]: - if cids is not None and cat["id"] not in cids_to_merge: + if cids is not None and annot["category_id"] not in cids_to_merge: continue annot["id"] = current_annot_id current_annot_id += 1