TypeError: индексы слайса должны быть целыми числами или None или иметь метод __index__ с нулевым заполнением CAGAN

Question

У меня есть ошибка по поводу TypeError: индексы срезов должны быть целыми числами или None или иметь метод index при запуске метода заполнения нулями в CA GAN (https://github.com/fei-hdu/ca-gan/tree/new_parsing)

Это код ошибки (в строке data.py 149)

def zero_padding(img, size0, pad1, pad2):
zero_padding = np.zeros((img.shape[0], size0, size0), dtype=np.float32)
print('img.shape[0] = ', img.shape[0])
pad1 = pad1 / 2
pad2 = pad2 / 2
zero_padding[:, pad1:size0 - pad1, pad2:size0 - pad2] = img
return zero_padding

Это ошибка

zero_padding [:, pad1: size0 - pad1, pad2: size0 - pad2] = img TypeError: индексы срезов должны быть целыми числами или отсутствовать или иметь index метод

Answer 1

Это ответ ..

Я только что получил это ..

def zero_padding(img, size0, pad1, pad2):
zero_padding = np.zeros((img.shape[0], size0, size0), dtype=np.float32)
pad1 = int(pad1 / 2)
pad2 = int(pad2 / 2)
zero_padding[:, pad1:size0 - pad1, pad2:size0 - pad2] = img
return zero_padding

Спасибо