ML based Motion Tracking and Synthetic MR Motion Image Generator
tagsim/
contains code used to generate motion images, and perform a Bloch
simulation to create MR images with proper contrast/features.
torch_tag/
contains the pyTorch implementation of the tracking network and
code used to train the network.
Software to generate pseudo random deformation MR images. Includes full image deformations and cardiac-like deformations, as well as a GPU accelerated Bloch simulator to generate MR images.
Demo Jupyter notebooks are in the tagsim/notebooks
folder.
Installation: This code requires a C library to be built for gridding. Running python setup.py build_ext --inplace
in the tagsim folder should build everything. If you are using XCode to on Mac for C compiling, replace setup.py
with setup_xcode.py
(this disables openMP because stock Mac XCode doesn't support it).
Software containing the neural network for tracking MR images. The full network implementation and pre-trained network are included, as well as a demo of its usage on an example dataset. Machine learning is implemented with pyTorch.
Demo Jupyter notebooks are in the torch_track/notebooks
folder.
A pre-train network for grid-tagged tracking is in the tagtorch_tracksim/network_saves
folder.
Installation: No special installation is required for this software, other than installing required dependencies as they come up.