Pyro: простой пример обратной графики с использованием SVI не работает - PullRequest
0 голосов
/ 15 февраля 2019

Я новичок в пиро и пытаюсь реализовать простую задачу обратной графики, включающую оценку координат точек треугольника, отображаемых на черно-белом изображении 32x32.

Итак, я определил генеративную модель, котораягенерирует 3 равномерно случайных точки, отображает их в изображение и наблюдает за результатом.Затем я использую SVI с автогидом (AutoMultivariateNormal), чтобы попытаться оценить точки для изображения с фиксированным треугольником.SVI, кажется, работает хорошо, и ELBO потери уменьшаются, однако, когда я пытаюсь произвести выборку сзади, все, что я получаю, это равномерно случайные точки без признаков обучения.

Мой код в блокноте Jupyter с результатами:

Screenshot #1

Screenshot #2

Screenshot #3

Что мне здесь не хватает?

...