pip install matplotlib

import json
from collections import Counter
import matplotlib.pyplot as plt
def analyze_coco_json(file_path):
    print(f"正在加載并分析文件: {file_path} ...\n")
    with open(file_path, 'r', encoding='utf-8') as f:
        coco_data = json.load(f)
    # 1. 基礎信息提取
    images = coco_data.get('images', [])
    annotations = coco_data.get('annotations', [])
    categories_info = coco_data.get('categories', [])
    num_images = len(images)
    num_annotations = len(annotations)
    categories = {cat['id']: cat['name'] for cat in categories_info}
    print("=== 1. 數據集基礎統(tǒng)計 ===")
    print(f"圖片總數: {num_images}")
    print(f"標注框總數: {num_annotations}")
    print(f"類別映射字典: {categories}\n")
    # 2. 統(tǒng)計各類別標注數量
    cat_counts = Counter([ann['category_id'] for ann in annotations])
    cat_distribution = {categories.get(k, f"Unknown ({k})"): v for k, v in cat_counts.items()}
    print("=== 2. 類別分布情況 ===")
    # 按數量從多到少排序
    sorted_cats = dict(sorted(cat_distribution.items(), key=lambda item: item[1], reverse=True))
    for cat, count in sorted_cats.items():
        print(f"  [{cat}]: {count} 個標注")
    # 3. 統(tǒng)計圖片分辨率
    resolutions = Counter([(img['width'], img['height']) for img in images])
    print("\n=== 3. 圖片分辨率分布 ===")
    for res, count in sorted(resolutions.items(), key=lambda x: x[1], reverse=True):
        print(f"  {res[0]}x{res[1]}: {count} 張圖片")
    # 4. 繪制類別分布柱狀圖并保存
    plt.figure(figsize=(8, 5))
    plt.bar(sorted_cats.keys(), sorted_cats.values(), color='#4C72B0')
    plt.title('Annotation Count per Category in COCO Dataset')
    plt.xlabel('Categories')
    plt.ylabel('Number of Annotations (BBoxes)')
    plt.xticks(rotation=45)
    # 在柱狀圖上顯示具體數值
    for i, (cat, count) in enumerate(sorted_cats.items()):
        plt.text(i, count + sum(sorted_cats.values())*0.01, str(count), ha='center')
    plt.tight_layout()
    plot_filename = 'category_distribution.png'
    plt.savefig(plot_filename, dpi=300)
    print(f"\n[成功] 類別分布可視化圖表已保存為: '{plot_filename}'")
if __name__ == "__main__":
    # 請確保將此處文件名替換為您實際的文件路徑
    analyze_coco_json('_annotations.coco.json')

python analyze_coco.py

plt.rcParams['font.sans-serif'] = ['SimHei']  # 用來正常顯示中文標簽
plt.rcParams['axes.unicode_minus'] = False   # 用來正常顯示負號