Skip to content
Snippets Groups Projects
Commit 2644e1c3 authored by David Johansen's avatar David Johansen
Browse files

fixes in generate_data.py

parent de6de083
No related branches found
No related tags found
No related merge requests found
......@@ -39,10 +39,10 @@ def main(output_dir: str, size: int):
vol_t = torch.from_numpy(cowry).to('cuda')
del cowry
size = 1000
target_shape = (size, size, size)
zoom_t = F.interpolate(vol_t.view(1, 1, *vol_t.shape), size=target_shape, mode='trilinear').view(*target_shape)
del vol_t
scale = 1000/size
ag = AcquisitionGeometry.create_Cone3D(
......@@ -66,7 +66,6 @@ def main(output_dir: str, size: int):
PO = ProjectionOperator(image_geometry=ig, acquisition_geometry=ag)
data = PO.direct(image)
output_dir = 'out/sim'
save_proj_as_tiffs(data, output_dir)
with open(os.path.join(output_dir, 'ag.pkl'), 'wb') as f:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment