Python TypeError: списочные индексы должны быть целыми или кусочками, а не кортежем - PullRequest
1 голос
/ 08 апреля 2020

Я пытаюсь извлечь объекты из изображения img с помощью свертки.

img_copy = np.copy(img)
x = img_copy.shape[0]
y = img_copy.shape[1]

matrix = [[-1, -2, -1], [0, 0, 0], [1, 2, 1]] # convolution matrix
weight = 1

def conv(x, y):
    val = 0.0
    for row,i in enumerate([-1, 0, 1]):
        for col,j in enumerate([-1, 0, 1]):
            val = val + img[x+j, y+i]*matrix[row, col]
    val = val*weight

    return val

for i in range(1, x-1):
    for j in range(1, y-1):
        pixel = conv(i, j)
        if(pixel<0):
            pixel = 0
        if(pixel>255):
            pixel = 255

При выполнении этого блока кода выдается следующая ошибка:

TypeError                                 Traceback (most recent call last)
<ipython-input-66-84eb09b3a0b7> in <module>
      1 for i in range(1, x-1):
      2     for j in range(1, y-1):
----> 3         pixel = conv(i, j)
      4         if(pixel<0):
      5             pixel = 0

<ipython-input-65-88737f90ffac> in conv(x, y)
      6     for row,i in enumerate([-1, 0, 1]):
      7         for col,j in enumerate([-1, 0, 1]):
----> 8             val = val + img[x+j, y+i]*matrix[row, col]
      9     val = val*weight
     10 

TypeError: list indices must be integers or slices, not tuple

Любая помощь, чтобы решить эту проблему проблема приветствуется.

Ответы [ 2 ]

1 голос
/ 08 апреля 2020

Ваш matrix является не массивом numpy, а списком Python, поэтому matrix[row, col] не может быть выполнен.

Таким образом, вы должны преобразовать это в numpy массив:

matrix = <b>np.array(</b>[[-1, -2, -1], [0, 0, 0], [1, 2, 1]]<b>)</b>
1 голос
/ 08 апреля 2020
val = val + img[x+j, y+i]*matrix[row, col]

Два значения, разделенные запятой, обычно интерпретируются как кортеж. Попробуйте вместо этого

val = val + img[x+j][y+i]*matrix[row][col]
...