IndexError: индекс 1 выходит за пределы оси 0 с размером 1 (python, mxnet) - PullRequest
Новая
       16

IndexError: индекс 1 выходит за пределы оси 0 с размером 1 (python, mxnet)

0 голосов
/ 09 января 2019

Я работаю в Mxnet для классификации изображений. Я загрузил набор данных с классом RecordFileDetection. Я хочу напечатать форму данных и метки. Но это показывает ошибку.

Мой сегмент кода: 

  from gluoncv.data import RecordFileDetection
  train_data_loader=RecordFileDetection('/home/vj/
                    Desktop/example_rec.rec',coord_normalized=True)
  print(len(train_data_loader))
  data =train_data_loader[0][1]
  print(data.shape)

Traceout

939 

 IndexError                                Traceback (most recent call 
 last)
 <ipython-input-70-3bc3bb8bd0b5> in <module>()
  2 
 train_data_loader=RecordFileDetection('/home/vj/
 Desktop/example_rec.rec',coord_normalized=True)
       3 print(len(train_data_loader))
 ----> 4 data =train_data_loader[0][1]
       5 print(data.shape)

 /usr/local/lib/python2.7/dist-packages/gluoncv/data/
recordio/detection.pyc in __getitem__(self, idx)
      71         h, w, _ = img.shape
      72         if self._coord_normalized:
 ---> 73             label = _transform_label(label, h, w)
      74         else:
      75             label = _transform_label(label)

 /usr/local/lib/python2.7/dist- 
 packages/gluoncv/data/recordio/detection.pyc in 
 _transform_label(label, height, width)
       8     label = np.array(label).ravel()
       9     header_len = int(label[0])  # label header
 ---> 10     label_width = int(label[1])  # the label width for each 
                                                object, >= 5
      11     if label_width < 5:
      12         raise ValueError(

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

Я не понимаю, в чем ошибка. Пожалуйста, помогите мне.

Заранее спасибо

...