IndexError: list index out of rangeを解決したい

Question

前提・実現したいこと

Mask R-CNNで自作データセットを学習させようとしていますが、エラーの解決方法がわからないです。

発生している問題・エラーメッセージ

画像二枚目のところにエラーコードがのっています。

以下がエラー先のvisualized.pyです

def display_top_masks(image, mask, class_ids, class_names, limit=4):
    """Display the given image and the top few class masks."""
    to_display = []
    titles = []
    to_display.append(image)
    titles.append("H x W={}x{}".format(image.shape[0], image.shape[1]))
    # Pick top prominent classes in this image
    unique_class_ids = np.unique(class_ids)
    mask_area = [np.sum(mask[:, :, np.where(class_ids == i)[0]])
                 for i in unique_class_ids]
    top_ids = [v[0] for v in sorted(zip(unique_class_ids, mask_area),
                                    key=lambda r: r[1], reverse=True) if v[1] > 0]
    Generate images and titles
    for i in range(limit):
       class_id = top_ids[i] if i < len(top_ids) else -1
        # Pull masks of instances belonging to the same class.
        m = mask[:, :, np.where(class_ids == class_id)[0]]
        m = np.sum(m * np.arange(1, m.shape[-1] + 1), -1)
        to_display.append(m)
        titles.append(class_names[class_id] if class_id != -1 else "-")
    display_images(to_display, titles=titles, cols=limit + 1, cmap="Blues_r")

該当のソースコード

ソースコード

試したこと

調べてvisualized.pyを変更したりしましたが、何も変化がありませんでした。

補足情報（FW/ツールのバージョンなど）

学習させたい画像の名前などです。

初投稿なので分かりにくいところがあると思いますが、宜しくお願い致します

Comments

[1T2R3M4]

https://qiita.com/k-ono0422/questions/7ee282ee013603fbb01d

以下ご対応ください。
https://teratail.com/help#posted-otherservice

Answer 1

class_names[class_id]でIndexError: list index out of rangeが出るということは、
len(class_names)<limitだということです。
その原因は提示された情報からはわかりませんので、ご自身で調べてください。