Plot gaussians

In [1]:

import matplotlib.pyplot as plt
import torch
from torch_grid_utils.coordinate_grid import coordinate_grid

from torch_find_peaks.gaussians import Gaussian2D, WarpedGaussian2D

g1 = Gaussian2D(1.,0., 0., 10., 5.)

grid = coordinate_grid((100,100), center=torch.tensor([50.,50.]))

im_g1 = g1(grid)

import matplotlib.pyplot as plt import torch from torch_grid_utils.coordinate_grid import coordinate_grid from torch_find_peaks.gaussians import Gaussian2D, WarpedGaussian2D g1 = Gaussian2D(1.,0., 0., 10., 5.) grid = coordinate_grid((100,100), center=torch.tensor([50.,50.])) im_g1 = g1(grid)

In [2]:

plt.imshow(im_g1.detach().numpy(), cmap='gray')

plt.imshow(im_g1.detach().numpy(), cmap='gray')

Out[2]:

<matplotlib.image.AxesImage at 0x7f080132da60>

In [6]:

g2 = WarpedGaussian2D(1.,0.,0., 5.0, 5.0, 0.2, 0.)

g2 = WarpedGaussian2D(1.,0.,0., 5.0, 5.0, 0.2, 0.)

In [7]:

im_g2 = g2(grid)[0]
plt.imshow(im_g2.detach().numpy(), cmap='gray')

im_g2 = g2(grid)[0] plt.imshow(im_g2.detach().numpy(), cmap='gray')

Out[7]:

<matplotlib.image.AxesImage at 0x7f2e51be4a10>

In [ ]:

In [ ]: