pytorch 発展ディープラーニング IndexError: index 4 is out of bounds for axis 1 with size 2

Question

失礼いたします。
pytorchの発展ディープラーニングを行っているのですがp74で記載通りコードを書いたのですが、インデックスエラーが出ました。
こちらどのようにすれば解決できますでしょうか？

コード
```phase="train"
img_transformed,boxex,labels=transform(
    img,phase,anno_list[:,:4],anno_list[:,4])
plt.imshow(cv2.cvColor(img_transformed,cv2_COLOR_BGR2RGB))
plt.show()

phase="val"
img_transformed,boxex,labels=transform(img,phase,anno_list[:,:4],anno_list[:,4])
plt.show(cv2.cvtColor(img_transformed,cv2.COLOR_GR2RGB))

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-146-698154b417cc> in <module>
      1 phase="train"
      2 img_transformed,boxex,labels=transform(
----> 3     img,phase,anno_list[:,:4],anno_list[:,4])
      4 plt.imshow(cv2.cvColor(img_transformed,cv2_COLOR_BGR2RGB))
      5 plt.show()

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

Answer 1

anno_listについてのエラーらしいです。
一度、anno_list.shape や anno_listを出力してみるなどして、配列の形状を確認してください。

ちなみに、

list=[
[1 ,2 ,3 ,4 ,5 ],
[7 ,8 ,9 ,10,11],
[13,14,15,16,17]
]

このような配列があったとき、
list[:,:4] は

[
[1 ,2 ,3 ,4 ],
[7 ,8 ,9 ,10],
[13,14,15,16]
]

という配列を返し、
list[:,4]は、

[[5],[11],[17]]

という配列を返します。(間違っていたらごめんなさい)