CambridgeCIA / LION

Learned Iterative Optimization Networks
GNU General Public License v3.0
20 stars 15 forks source link

Add FBPConvNet test demo script. Fix typos. Fix FBPConvNet_train.py. #106

Closed cshoebridge closed 5 months ago

cshoebridge commented 5 months ago

FBPConvNet_test.py - small test script to show trained models performance on unseen data. Largely for visual purposes. In future may create a more generic tool for this.

FBPConvNet_train.py - removed redundant imports and fixed outdated imports - Parameter in place of LIONParameter seems to appear a lot throughout the repo. Fixed test logic, fdk was applied to data before feeding it into FBPConvNet, which already applied fdk internally; FBPConvNet expects a sinogram.