オリジナルデータでSSD実行途中

Question

https://github.com/qfgaohao/pytorch-ssd
こちらをもとに、オリジナルデータでの学習をさせようとしています。

【行ったこと】
・オリジナルデータは
画像はリサイズをし、500×281pxにしました。
そしてvottでアノテーションし、voc形式で出力後、座標の小数点以下を消しました。
segmentationclassとsegmentationobjectのフォルダは作っていません。

・コードは
pytorch-ssd-master\vision\datasets\voc_dataset.pyのclass_nameにアノテーションのラベルを追記しました（else文中で意味ないのかもしれませんが）

1番下のエラーが解決できません。
どうぞアドバイスをお願いいたします。

実行コマンド

py train_ssd.py --datasets /Users/../Desktop/AI/original_datasets_0925_test/pytorch-ssd-master/data/200930/BtoT_ng  --validation_dataset /Users/../Desktop/AI/original_datasets_0925_test/pytorch-ssd-master/data/200930/BtoT_ng --net mb2-ssd-lite --base_net models/mb2-imagenet-71_8.pth --scheduler cosine --lr 0.01 --t_max 200 --validation_epochs 5 --num_epochs 20 --num_workers 0

エラー内容

 File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\datasets\voc_dataset.py", line 65, in __getitem__
    image, boxes, labels = self.transform(image, boxes, labels)
  File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\ssd\data_preprocessing.py", line 34, in __call__
    return self.augment(img, boxes, labels)
  File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\transforms\transforms.py", line 55, in __call__
    img, boxes, labels = t(img, boxes, labels)
  File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\transforms\transforms.py", line 344, in __call__
    boxes[:, :2] += (int(left), int(top))
IndexError: too many indices for array

Comments

[meg_]

boxesのshapeはどうなっていますか？

[ken248]

(0,)　　となります。
アノテーションが認識されていないということでしょうか・・・？？

[meg_]

中身がないってことでしょうか？ 何が原因かはコードをデバッグしないと分からないかと。

Answer 1

今回は断念しました。

Answer 2

IndexError: 配列のインデックスが多すぎます

とおっしゃってますよ

Comments

[ken248]

boxes.shapeで(0,)
となるのですが…それでも多すぎるのでしょうか？

意味が理解できないです。
もしよろしければ教えていただきたいです。

[y_waiwai]

コードが提示されてないので、説明しようにもどうにもできません

[ken248]

train_ssd.pyもtransforms.pyも
コード全文をのせようとすると文字数オーバーになってしまいます。。
申し訳ございません。