mirror of
https://github.com/cdtools-developers/cdtools.git
synced 2026-09-09 21:12:42 +02:00
27 lines
766 B
Python
27 lines
766 B
Python
import cdtools
|
|
from matplotlib import pyplot as plt
|
|
|
|
# We load an example dataset from a .cxi file
|
|
filename = 'example_data/lab_ptycho_data.cxi'
|
|
dataset = cdtools.datasets.Ptycho2DDataset.from_cxi(filename)
|
|
|
|
# We create a ptychography model from the dataset
|
|
model = cdtools.models.SimplePtycho.from_dataset(dataset)
|
|
|
|
# We move the model to the GPU
|
|
device = 'cuda'
|
|
model.to(device=device)
|
|
dataset.get_as(device=device)
|
|
|
|
# We run the actual reconstruction
|
|
for loss in model.Adam_optimize(100, dataset, batch_size=10):
|
|
# We print a quick report of the optimization status
|
|
print(model.report())
|
|
# And liveplot the updates to the model as they happen
|
|
model.inspect(dataset)
|
|
|
|
# We study the results
|
|
model.inspect(dataset)
|
|
model.compare(dataset)
|
|
plt.show()
|