pycocotools

import os from pycocotools.coco import COCO # NOTE: You need to download COCO annotations first. # For example, 'instances_val2017.json' can be found at: # http://images.cocodataset.org/annotations/annotations_trainval2017.zip # Placeholder for annotation file path. Replace with actual path. annFile = os.environ.get('COCO_ANN_FILE', 'path/to/instances_val2017.json') # Initialize COCO API for instance annotations try: coco = COCO(annFile) print(f"Successfully loaded COCO annotations from {annFile}") # Get all image IDs imgIds = coco.getImgIds() print(f"Total images: {len(imgIds)}") # Get category IDs catIds = coco.getCatIds(catNms=['person', 'dog', 'cat']) print(f"Categories found: {coco.loadCats(catIds)}") # Pick an image and display some info img = coco.loadImgs(imgIds[0])[0] print(f"First image info: {img}") # Get annotations for the image annIds = coco.getAnnIds(imgIds=img['id'], catIds=catIds, iscrowd=None) anns = coco.loadAnns(annIds) print(f"Annotations for image {img['id']}: {len(anns)} found") except Exception as e: print(f"Error loading COCO annotations or processing: {e}") print("Please ensure the annotation file path is correct and accessible.")