neural-network - COCO データセット クラスあたりの画像数

Question

COCO2017 には 118k のトレーニングと 5k の検証データセット (122k の画像) の 80 クラスがあります。ここで質問があります。クラスあたりの画像数 (クラスあたり 1525 画像) は 122k / 80 ですか?

Answer 1

COCO データセットは均等に分散されたデータセットではありません。つまり、すべてのクラスが同じ数の画像を持っているわけではありません。そこで、任意のクラスの画像数を調べる方法を紹介しましょう。

PyCoco APIを使用して、COCO データセットを操作しています。COCO データセットの「人物」クラスの画像数を調べてみましょう。COCO データセットから任意のクラスを除外するコードの要点を次に示します。

# Define the class (out of the 80 COCO classes)
filterClasses = ['person']

# Fetch class IDs only corresponding to the filterClasses
catIds = coco.getCatIds(catNms=filterClasses) 

# Get all images containing the above Category IDs
imgIds = coco.getImgIds(catIds=catIds)
print("Number of images containing the class:", len(imgIds))

そこで、データセット内の「人」に対応する画像の数を取得します!

私は最近、COCO データセットの探索と操作に関する投稿全体を書きました。詳細とコード全体を確認してください。