Is there a pytorch version of the code?

ShouYuqing / rigid

Rigid registration for 3D MRI

GNU General Public License v3.0

12 stars 0 forks source link

Is there a pytorch version of the code? #2

Closed ljjiayou closed 12 months ago

ljjiayou commented 1 year ago

Is there a pytorch version of the code? Not familiar with tensorflow,I am looking for some code on deep learning 3D multi-modal rigid registration, which is hard to find.

ShouYuqing commented 12 months ago

Currently, all models are developed in TensorFlow, u can try to translate them into Pytorch.