Skip to content

r20220512

Latest
Latest
Compare
Choose a tag to compare
@EvanBeal EvanBeal released this 12 May 13:22
· 2 commits to main since this release
r20220512
9679cd4
This commit was created on github.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Trained cascaded models (PyTorch)

Registration models trained with the SynthMorph method and converted from .h5 (TensorFlow/Keras) to .pt (PyTorch). The two models provided in this release (in cascaded_models_pytorch.zip) are the same as the ones provided in the release r20220329.

The configuration files used to train these models originally are present in cascaded_models_pytorch.zip. And the files used to convert the models from TensorFlow/Keras to Pytorch are also included in cascaded_models_pytorch.zip

An example script + example data to test these models is included in cascaded-registration-example.zip. To test this example follow these steps:

  1. Download and extract cascaded-registration-example.zip.
  2. Download and extract cascaded_models_pytorch.zip to the model/ folder inside cascaded-registration-example/.
  3. Run the following commands:
cd cascaded-registration-example

python -m venv venv
source venv/bin/activate
pip install -r requirements.txt

python cascaded-registration.py
Assets 5
Loading