YOLOv5のIndexError

YOLOv5で画像の物体検出をしたいと思っています

実現したいこと

LabelImgで画像をアノテーションし、一つだけのクラスで分類しました

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

IndexError: index 1 is out of bounds for axis 0 with size 1

該当のソースコード

ソースコード

試したこと

初学者ですみませんが、どうすればこのエラーは消えますでしょうか。

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

ここにより詳細な情報を記載してください。