Skip to content

Commit

Permalink
Rerun examples that use the new projectors
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael-T-McCann committed Sep 11, 2024
1 parent 9bb7b68 commit ce1ed1e
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions examples/scripts/ct_projector_comparison_3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,10 @@
import jax
import jax.numpy as jnp

import scico.linop.xray
import scico.linop.xray.astra as astra
from scico import plot
from scico.examples import create_block_phantom
from scico.linop import XRayTransform, XRayTransform3D
from scico.linop.xray import XRayTransform3D
from scico.util import ContextTimer, Timer

"""
Expand Down Expand Up @@ -58,7 +57,7 @@

timer_scico = Timer()
with ContextTimer(timer_scico, "init"):
H_scico = XRayTransform(XRayTransform3D(in_shape, matrices, out_shape))
H_scico = XRayTransform3D(in_shape, matrices, out_shape)

with ContextTimer(timer_scico, "first_fwd"):
y_scico = H_scico @ x
Expand All @@ -84,9 +83,7 @@
Convert SCICO geometry to ASTRA and project.
"""

vectors_from_scico = scico.linop.xray.astra.convert_from_scico_geometry(
in_shape, matrices, out_shape
)
vectors_from_scico = astra.convert_from_scico_geometry(in_shape, matrices, out_shape)

timer_astra = Timer()
with ContextTimer(timer_astra, "init"):
Expand Down Expand Up @@ -132,7 +129,7 @@
Convert ASTRA geometry to SCICO and project.
"""

P_from_astra = scico.linop.xray.astra._astra_to_scico_geometry(H_astra.vol_geom, H_astra.proj_geom)
P_from_astra = astra._astra_to_scico_geometry(H_astra.vol_geom, H_astra.proj_geom)
H_scico_from_astra = XRayTransform3D(in_shape, P_from_astra, out_shape)

y_scico_from_astra = H_scico_from_astra @ x
Expand Down

0 comments on commit ce1ed1e

Please sign in to comment.