uncbiag / mermaid

Image registration using pytorch
Other
178 stars 29 forks source link

Clear model function for atlas builder #95

Closed StephanieHaeger closed 6 years ago

StephanieHaeger commented 6 years ago

The atlas builder runs svf registration in a for loop and uses model properties from the previous registration for the current registration. A function that clears the model beforehand is needed.

marcniethammer commented 6 years ago

Should be possible now. For the simple registration interface, one can now do the following:

import pyreg.simple_interface as si reg = si.RegisterImagePair() ...

reg.set_model_parameters_to_zero()

This should be done with care though, because it will set all model parameters to zero.