Как исправить ошибку "IndexError ('Стоп среза% d превышает предел% d'% (stop, length))" в Python - PullRequest
Новая
       37

Как исправить ошибку "IndexError ('Стоп среза% d превышает предел% d'% (stop, length))" в Python

0 голосов
/ 05 июня 2019

У меня была проблема, когда я применяю реализацию arcface из этого репо , я получил ошибку.

Из этой строки код: 

face_imgs_resized = np.array(face_imgs_resized)
face_imgs_resized = np.rollaxis(face_imgs_resized, 3, 1)

data = self.mx.nd.array(face_imgs_resized)
db = self.mx.io.DataBatch(data=(data,))
self.model.forward(db, is_train=False)

И ошибка: 

    self.model.forward(db, is_train=False)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/module/module.py", line 625, in forward
    self.exec_group.forward(data_batch, is_train)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/module/executor_group.py", line 450, in forward
    load_data(data_batch, self.data_arrays, self.data_layouts)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/module/executor_group.py", line 74, in _load_data
    _load_general(batch.data, targets, major_axis)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/module/executor_group.py", line 48, in _load_general
    d_src[slice_idx.start:slice_idx.stop].copyto(d_dst)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/ndarray/ndarray.py", line 506, in __getitem
    return self._get_nd_basic_indexing(key)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/ndarray/ndarray.py", line 787, in _get_nd_basic_indexing
    return self._slice(key.start, key.stop)
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/ndarray/ndarray.py", line 902, in _slice
    start, stop, _ = _get_index_range(start, stop, self.shape[0])
  File "/root/miniconda/envs/roy/lib/python3.7/site-packages/mxnet/ndarray/ndarray.py", line 2327, in _get_index_range
    raise IndexError('Slicing stop %d exceeds limit of %d' % (stop, length))
IndexError: Slicing stop 2 exceeds limit of 1

Так кто-нибудь когда-нибудь встречал эту ошибку раньше? И, пожалуйста, покажи мне, как это исправить.

1 Ответ

0 голосов
/ 04 июля 2019

Вы передаете NDArray в DataBatch, когда DataBatch принимает массив NDArrays, каждый из которых имеет размер batch_size.

Можете ли вы попробовать это? 

face_imgs_resized = np.ones((128,224,224,3))
face_imgs_resized = np.array(face_imgs_resized)
face_imgs_resized = np.rollaxis(face_imgs_resized, 3, 1)
data = mx.nd.array(face_imgs_resized)
db = mx.io.DataBatch(data=mx.nd.split_v2(data, 4))
self.model.forward(db, is_train=False)
...