BloodAxe/pytorch-toolbelt

Ошибка при вызове метода merge ?

Vova-B opened this issue · 0 comments

Спасибо за удобные инструменты для работы но я не совсем понял в каком виде я должен передавать tiles в метод merge. У меня tiles выглядит как [[512, 512, 1], [512, 512, 1], ...] и когда я вызываю для них метод merge я получаю такую ошибку:

image[y:y + tile_height, x:x + tile_width] += tile * w
ValueError: non-broadcastable output operand with shape (512,512,1) doesn't match the broadcast shape (512,512,512)

Не подскажите в чем может быть вызвана эта проблема ?