Объект «Изображение» не вызывается
Я играл со стабильной диффузией в Google Colab, и у меня возникли проблемы с работой сетки. Каждый раз, когда я запускаю свой блок генератора изображений, он проходит генерацию, но не может отображать изображения, выдает мне ошибку, указанную в заголовке.
def image_grid(imgs, rows, cols):
assert len(imgs) == rows*cols
w, h = imgs[0].size
grid = Image.new('RGB', size=(cols*w, rows*h))
grid_w, grid_h = grid.size
for i, img in enumerate(imgs):
grid.paste(img, box=(i%cols*w, i//cols*h))
return grid
Это работает нормально, но следующая часть меня заводит.
from torch import autocast
num_images = 4
prompt = ["Astronaut riding a horse"] * num_images
with autocast("cuda"):
images = pipe(prompt, width=512, height=512, num_inference_steps=10)["sample"]
grid = image_grid(images, rows=2, cols=2)
grid
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-30-ec9ce225f081> in <module>
16 images = pipe(prompt, width=512, height=512, num_inference_steps=10)["sample"]
17
---> 18 grid = image_grid(images, rows=2, cols=2)
19 grid
<ipython-input-28-2e81f9e2985a> in image_grid(imgs, rows, cols)
10 for i, img in enumerate(imgs):
11 grid.paste(img, box=(i%cols*w, i//cols*h))
---> 12 return grid()
TypeError: 'Image' object is not callable
1 ответ
Сначала удалил [0], затем изменил «возвратную сетку ()» на «возвратную сетку». Спасибо, OneCricketeer.