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
エラー内容
python
1 File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\datasets\voc_dataset.py", line 65, in __getitem__ 2 image, boxes, labels = self.transform(image, boxes, labels) 3 File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\ssd\data_preprocessing.py", line 34, in __call__ 4 return self.augment(img, boxes, labels) 5 File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\transforms\transforms.py", line 55, in __call__ 6 img, boxes, labels = t(img, boxes, labels) 7 File "C:\Users..\Desktop\AI\original_datasets_0925_test\pytorch-ssd-master\vision\transforms\transforms.py", line 344, in __call__ 8 boxes[:, :2] += (int(left), int(top)) 9IndexError: too many indices for array
回答2件
あなたの回答
tips
プレビュー