search

DeepRegNet / DeepReg

Medical image registration using deep learning
Apache License 2.0
564 stars 76 forks source link

Move BSpline functions outside of backbone #645

Closed mathpluscode closed 3 years ago

mathpluscode commented 3 years ago

Subject of the issue

As mentioned in https://github.com/DeepRegNet/DeepReg/pull/539#issuecomment-766143980, the functionalities related to Bsplined should be a backbone-independent layer, instead of being hardcoded in backbones, https://github.com/DeepRegNet/DeepReg/blob/main/deepreg/model/backbone/u_net.py#L76.

Therefore refactoring is required as we may have more backbones in the future.

acasamitjana commented 3 years ago

I have removed the resize+Bsplines from backbone networks and added into the ddf and dvf models. I updated the tests accordingly.