Skip to content

Instantly share code, notes, and snippets.

@kipavy

kipavy/pyvsnr_prof.py

Created June 17, 2024 13:43
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save kipavy/413e2ac4f1b3ab5daa898a4a333a0249 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save kipavy/413e2ac4f1b3ab5daa898a4a333a0249 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pyvsnr_prof.py
# https://github.com/CEA-MetroCarac/pyvsnr/tree/data-loading-enhancement
# cProfile.Profile()
import cupy as cp
import cupyx
from pyvsnr import vsnr2d
filters=[{'name':'Dirac', 'noise_level':0.35}]
img = cp.random.rand(2048, 2048).astype(cp.float32)
nit=20
with cupyx.profiler.profile():
for i in range(nit):
vsnr2d(img, filters)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment