GradICON requires the itk_wrapper

uncbiag / ICON

A library for performing image registration using deep learning, regularized by inverse consistency

Other

42 stars 9 forks source link

GradICON requires the itk_wrapper #66

Closed curiale closed 4 months ago

curiale commented 11 months ago

GradICON requires itk_wrapper.

File ".../GradICON_Det/determinant_helper.py", line 276, in register_pair_mask icon.itk_wrapper.create_itk_transform(phi_AB, model.identity_map, image_A, image_B), AttributeError: module 'icon_registration' has no attribute ‘itk_wrapper’

HastingsGreer commented 10 months ago

I think that this is a good idea. The issue is that itk is extremely slow to import, so we want to avoid loading it when icon_registration is imported for a non-itk task, just to improve developer experience. Maybe we could also move the itk import into the register_pair function body?