😸

IndexError: index 1 is out of bounds for dimension 1 with size 1解決方法

2023/09/05に公開

エラーIndexError: index 1 is out of bounds for dimension 1 with size 1

原因

ラベルのIDと設定ファイルのクラス数(nc)の不一致
ラベルに ID 0 と 1 があり、構成で nc=1 を設定している場合、このエラーが発生します。

解決方法

検出したいTargetが一つだけの場合は、data.yamlのncを1にする。

train: ./train/images
val: ./valid/images

nc: 1
names: ['surfer']

そしてラベルのクラスID(一番左の値)が全て0であることを確認する

0 0.480273 0.328488 0.259766 0.647287

その後、labels.cacheを削除し、再実行すればエラー解決!

参考

https://github.com/ultralytics/ultralytics/issues/472

Discussion