from pySART import pysart
from skimage.transform import radon, iradon
import phantom
import numpy as np

def generate_phantom():
    theta = range(0,180)
    data = np.rot90(radon(phantom.phantom(n=100),theta=theta))
    data1 = np.zeros((data.shape[0],1,data.shape[1]))
    data1[:,0,:] = data
    return theta, data1

pysart = pysart(*generate_phantom())
pysart.sart()

pysart = pysart(*generate_phantom(), plsice = 10, iteration=100)
pysart.sart()