New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Parallelized affine registration #1608

Closed
wants to merge 6 commits into
base: master
from

Conversation

Projects
None yet
2 participants
@RicciWoo
Copy link
Contributor

RicciWoo commented Aug 6, 2018

Parallelized affine registration using prange and local function.

  1. cythonized _gradient_* and _sparse_gradient_* in vector_fields.pyx;
  2. typed and made nogil on _compute_pdfs_* and _joint_pdf_gradient_* in parzenhist.pyx;
  3. parallelize _gradient_3d using prange and local function.

To specify number of threads, add num_threads when initializing MutualInformationMetric:

metric = MutualInformationMetric(nbins, sampling_prop, num_threads=4)

@skoudoro skoudoro added the gsoc2018 label Aug 6, 2018

@RicciWoo RicciWoo force-pushed the RicciWoo:aff-par-hist-grad-thr branch from 1fa80ff to 4078a3a Aug 8, 2018

@RicciWoo RicciWoo closed this Aug 9, 2018

@RicciWoo RicciWoo deleted the RicciWoo:aff-par-hist-grad-thr branch Aug 10, 2018

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment